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
46 friend class ScopedAllocatorMgr;
47 ScopedAllocatorContainer(const ScopedAllocatorMgr* mgr, int64 step_id) in ScopedAllocatorContainer()
52 const ScopedAllocatorMgr* mgr_;
73 class ScopedAllocatorMgr {
75 explicit ScopedAllocatorMgr(const string& device_name) in ScopedAllocatorMgr() function
77 ~ScopedAllocatorMgr();
|
| D | scoped_allocator_mgr.cc | 117 ScopedAllocatorMgr::~ScopedAllocatorMgr() { in ~ScopedAllocatorMgr()
130 void ScopedAllocatorMgr::Cleanup(int64 step_id) { in Cleanup()
139 ScopedAllocatorContainer* ScopedAllocatorMgr::GetContainer(int64 step_id) { in GetContainer()
153 Status ScopedAllocatorMgr::AddScopedAllocator( in AddScopedAllocator()
164 size_t ScopedAllocatorMgr::PopulateFields( in PopulateFields()
|
| D | threadpool_device.h | 35 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr()
46 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 | 51 scoped_allocator_mgr_(new ScopedAllocatorMgr(name)) { in ThreadPoolDevice()
|
| D | renamed_device.h | 67 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr()
|
| D | direct_session.cc | 1650 ScopedAllocatorMgr* sam = d->GetScopedAllocatorMgr(); in RunState()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 50 class ScopedAllocatorMgr; variable
202 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 | 87 ScopedAllocatorMgr* sam, const string& op_name, in PrepOp()
91 ScopedAllocatorMgr::PopulateFields(id, fields_shapes, dtype, fields); in PrepOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.h | 115 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr()
135 std::unique_ptr<ScopedAllocatorMgr> scoped_allocator_mgr_;
|
| D | gpu_device.cc | 309 scoped_allocator_mgr_(new ScopedAllocatorMgr(name)), in BaseGPUDevice()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker.cc | 355 ScopedAllocatorMgr* sam = d->GetScopedAllocatorMgr(); in CleanupGraphAsync()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 303 int64 num_bytes = ScopedAllocatorMgr::PopulateFields( in AnalyzeInputs()
|