Home
last modified time | relevance | path

Searched refs:ScopedAllocator (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dscoped_allocator_mgr.h37 const gtl::ArraySlice<ScopedAllocator::Field>& fields,
41 ScopedAllocator* GetAllocator(int32 scope_id);
44 void Drop(int32 scope_id, ScopedAllocator* sa);
59 ScopedAllocator* scoped_allocator;
64 SAField(int32 fi, ScopedAllocator* sa) in SAField()
67 : field_index(ScopedAllocator::kBackingIndex), in SAField()
86 const gtl::ArraySlice<ScopedAllocator::Field>& fields,
99 std::vector<ScopedAllocator::Field>* fields);
Dscoped_allocator_mgr.cc24 const gtl::ArraySlice<ScopedAllocator::Field>& fields, in AddScopedAllocator()
44 ScopedAllocator* sa = new ScopedAllocator( in AddScopedAllocator()
47 ScopedAllocatorContainer::SAField(ScopedAllocator::kBackingIndex, sa); in AddScopedAllocator()
50 const ScopedAllocator::Field& f = fields[i]; in AddScopedAllocator()
59 ScopedAllocator* ScopedAllocatorContainer::GetAllocator(int32 scope_id) { in GetAllocator()
63 CHECK_EQ(ScopedAllocator::kBackingIndex, it->second.field_index); in GetAllocator()
86 void ScopedAllocatorContainer::Drop(int32 scope_id, ScopedAllocator* sa) { in Drop()
92 if (it->second.field_index != ScopedAllocator::kBackingIndex) { in Drop()
108 if (it.second.field_index == ScopedAllocator::kBackingIndex) { in ~ScopedAllocatorContainer()
155 const gtl::ArraySlice<ScopedAllocator::Field>& fields, in AddScopedAllocator()
[all …]
Dscoped_allocator.cc22 ScopedAllocator::ScopedAllocator(const Tensor& backing_tensor, int32 scope_id, in ScopedAllocator() function in tensorflow::ScopedAllocator
42 ScopedAllocator::~ScopedAllocator() { in ~ScopedAllocator()
58 void* ScopedAllocator::AllocateRaw(int32 field_index, size_t num_bytes) { in AllocateRaw()
124 void ScopedAllocator::DeallocateRaw(void* p) { in DeallocateRaw()
142 bool ScopedAllocator::VerifyPointer(const void* p) { in VerifyPointer()
157 bool ScopedAllocator::VerifyTensor(const Tensor* t) { in VerifyTensor()
161 ScopedAllocatorInstance::ScopedAllocatorInstance(ScopedAllocator* sa, in ScopedAllocatorInstance()
Dscoped_allocator.h28 class ScopedAllocator {
47 ScopedAllocator(const Tensor& backing_tensor, int32 scope_id,
54 ~ScopedAllocator() TF_LOCKS_EXCLUDED(mu_);
90 explicit ScopedAllocatorInstance(ScopedAllocator* sa, int32 field_index);
118 ScopedAllocator* scoped_allocator_;
Dscoped_allocator_mgr_test.cc88 std::vector<ScopedAllocator::Field> fields_;
149 ScopedAllocator* other = sac->GetAllocator(scope_id_); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dscoped_allocator_ops_test.cc86 std::vector<ScopedAllocator::Field>* fields, in PrepOp()
172 std::vector<ScopedAllocator::Field> fields; in ExecOp()
291 std::vector<ScopedAllocator::Field> fields; in ExecOp()
Dscoped_allocator_ops.cc79 std::vector<ScopedAllocator::Field> fields_;
/external/tensorflow/tensorflow/core/framework/
Dtensor.h680 friend class ScopedAllocator; // For access to buf_. variable
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc470 std::vector<ScopedAllocator::Field> sa_fields; in AnalyzeInputs()