Searched refs:ScopedAllocatorMgr (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator_mgr.h | 27 class ScopedAllocatorMgr; variable
47 friend class ScopedAllocatorMgr;
48 ScopedAllocatorContainer(const ScopedAllocatorMgr* mgr, int64 step_id) in ScopedAllocatorContainer()
53 const ScopedAllocatorMgr* mgr_;
74 class ScopedAllocatorMgr {
76 explicit ScopedAllocatorMgr(const std::string& device_name) in ScopedAllocatorMgr() function
78 ~ScopedAllocatorMgr();
|
| D | scoped_allocator_mgr.cc | 116 ScopedAllocatorMgr::~ScopedAllocatorMgr() { in ~ScopedAllocatorMgr()
129 void ScopedAllocatorMgr::Cleanup(int64 step_id) { in Cleanup()
138 ScopedAllocatorContainer* ScopedAllocatorMgr::GetContainer(int64 step_id) { in GetContainer()
152 Status ScopedAllocatorMgr::AddScopedAllocator( in AddScopedAllocator()
163 size_t ScopedAllocatorMgr::PopulateFields( in PopulateFields()
|
| D | threadpool_device.h | 35 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr()
49 std::unique_ptr<ScopedAllocatorMgr> scoped_allocator_mgr_;
|
| D | scoped_allocator_mgr_test.cc | 35 ScopedAllocatorMgr::PopulateFields(scope_id_, fields_shapes_, DT_FLOAT, in PopulateFields()
89 ScopedAllocatorMgr sam_;
|
| D | threadpool_device.cc | 52 scoped_allocator_mgr_(new ScopedAllocatorMgr(name)) { in ThreadPoolDevice()
|
| D | renamed_device.h | 66 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr()
|
| D | direct_session.cc | 1773 ScopedAllocatorMgr* sam = d->GetScopedAllocatorMgr(); in RunState()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 49 class ScopedAllocatorMgr; variable
201 virtual ScopedAllocatorMgr* GetScopedAllocatorMgr() const { return nullptr; } in GetScopedAllocatorMgr()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scoped_allocator_ops.cc | 41 ScopedAllocatorMgr::PopulateFields(id_, shapes_, dtype_, &fields_); in ScopedAllocatorOp()
51 ScopedAllocatorMgr* sam = device_->GetScopedAllocatorMgr(); in Compute()
|
| D | scoped_allocator_ops_test.cc | 88 ScopedAllocatorMgr* sam, const string& op_name, in PrepOp()
92 ScopedAllocatorMgr::PopulateFields(id, fields_shapes, dtype, fields); in PrepOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.h | 102 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr()
127 std::unique_ptr<ScopedAllocatorMgr> scoped_allocator_mgr_;
|
| D | gpu_device.cc | 390 scoped_allocator_mgr_(new ScopedAllocatorMgr(name)), in BaseGPUDevice()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker.cc | 360 ScopedAllocatorMgr* sam = d->GetScopedAllocatorMgr(); in CleanupGraphAsync()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 473 int64 num_bytes = ScopedAllocatorMgr::PopulateFields( in AnalyzeInputs()
|