| 39 | using ReleaserHookT = std::function<void(const std::shared_ptr<vm::TensorStorage>&)>; |
| 40 | |
| 41 | const std::shared_ptr<vm::TensorStorage> storage() const { return storage_; } |
| 42 | |
| 43 | void set_releaser_hook(const ReleaserHookT& releaser_hook) { |
| 44 | releaser_hook_ = std::make_shared<ReleaserHookT>(releaser_hook); |