You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.5 KiB
52 lines
1.5 KiB
8 years ago
|
#pragma once
|
||
|
|
|
||
|
|
#include <vector>
|
||
|
|
#include <unordered_map>
|
||
|
|
#include "paddle/framework/variable.h"
|
||
|
|
#include "paddle/utils/Error.h"
|
||
|
|
|
||
|
|
namespace paddle {
|
||
|
|
namespace framework {
|
||
|
|
|
||
|
|
const static Error AlreadyCreated("Variable has already been created");
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Scope is an association of a name to Variable. All variables belong to `Scope`.
|
||
|
|
* You need to specify a scope to run a Net, i.e., `net.Run(&scope)`. One net can
|
||
|
|
* run in different scopes and update different variable in the scope.
|
||
|
|
*/
|
||
|
|
class Scope {
|
||
|
|
public:
|
||
|
|
Scope() {}
|
||
|
|
|
||
|
|
explicit Scope(const std::shared_ptr<Scope> &scope):
|
||
|
|
parent_(scope) {}
|
||
|
|
|
||
|
|
~Scope() {}
|
||
|
|
|
||
|
|
// Create Variable in this Scope. Return error if Variable already been
|
||
|
|
// created.
|
||
|
|
Error __must_check CreateVariable(const std::string& name);
|
||
|
|
|
||
|
|
// Get Variable from this Scope, this function will recursive find Variable
|
||
|
|
// from it's parent scope.
|
||
|
|
// Return nullptr if not found.
|
||
|
|
Variable* GetVariable(const std::string& name) const;
|
||
|
|
|
||
|
|
// find and return Variables in the scope it self.
|
||
|
|
Variable* GetVarLocally(const std::string& name) const;
|
||
|
|
|
||
|
|
// Get a Variable from Scope, if the Variable is not exist then create it.
|
||
|
|
// User should call this function most of time.
|
||
|
|
Variable* GetOrCreateVariable(const std::string& name);
|
||
|
|
|
||
|
|
bool HaveVariable(const std::string& name);
|
||
|
|
|
||
|
|
private:
|
||
|
|
std::unordered_map<std::string, std::unique_ptr<Variable>> vars_;
|
||
|
|
std::shared_ptr<Scope> parent_ {nullptr};
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace framework
|
||
|
|
} // namespace paddle
|