"switch to shared_ptr"

gangliao-patch-1
dongzhihong 8 years ago
parent 18e65b0c08
commit b871641a53

@ -11,7 +11,7 @@ void PlainNet::InferShape(Scope* scope) {
}
}
void PlainNet::Run(Scope* scope, DeviceContext* ctx) {
void PlainNet::Run(std::shared_ptr<Scope> scope, DeviceContext* ctx) {
for (auto& op : ops_) {
op.Run(ctx);
}

@ -69,7 +69,7 @@ class Net {
* environment for ops. `begin` and `end` specify the scope of `ops_` to run,
* If no positive indexes are provided, all operators in `ops_` will run.
*/
virtual void Run(Scope *scope, DeviceContext *ctx) = 0;
virtual void Run(std::shared_ptr<Scope> scope, DeviceContext *ctx) = 0;
/**
* @brief Add an Operator according to `def`.
@ -123,7 +123,7 @@ class PlainNet : public Net {
* scope will be used instead. If no OpContext is provicded, default context
* will be used.
*/
virtual void Run(Scope *scope, DeviceContext *ctx) override;
virtual void Run(std::shared_ptr<Scope> scope, DeviceContext *ctx) override;
/**
* @brief Add an operator to this network.

Loading…
Cancel
Save