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.h36 const gtl::ArraySlice<ScopedAllocator::Field>& fields,
40 ScopedAllocator* GetAllocator(int32 scope_id);
43 void Drop(int32 scope_id, ScopedAllocator* sa);
58 ScopedAllocator* scoped_allocator;
63 SAField(int32 fi, ScopedAllocator* sa) in SAField()
66 : field_index(ScopedAllocator::kBackingIndex), in SAField()
85 const gtl::ArraySlice<ScopedAllocator::Field>& fields,
98 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()
109 if (it.second.field_index == ScopedAllocator::kBackingIndex) { in ~ScopedAllocatorContainer()
156 const gtl::ArraySlice<ScopedAllocator::Field>& fields, in AddScopedAllocator()
[all …]
Dscoped_allocator.cc20 ScopedAllocator::ScopedAllocator(const Tensor& backing_tensor, int32 scope_id, in ScopedAllocator() function in tensorflow::ScopedAllocator
40 ScopedAllocator::~ScopedAllocator() { in ~ScopedAllocator()
56 void* ScopedAllocator::AllocateRaw(int32 field_index, size_t num_bytes) { in AllocateRaw()
99 void ScopedAllocator::DeallocateRaw(void* p) { in DeallocateRaw()
117 bool ScopedAllocator::VerifyPointer(const void* p) { in VerifyPointer()
132 bool ScopedAllocator::VerifyTensor(const Tensor* t) { in VerifyTensor()
136 ScopedAllocatorInstance::ScopedAllocatorInstance(ScopedAllocator* sa, in ScopedAllocatorInstance()
Dscoped_allocator.h28 class ScopedAllocator {
46 ScopedAllocator(const Tensor& backing_tensor, int32 scope_id,
53 ~ScopedAllocator() LOCKS_EXCLUDED(mu_);
89 explicit ScopedAllocatorInstance(ScopedAllocator* sa, int32 field_index);
116 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.cc85 std::vector<ScopedAllocator::Field>* fields, in PrepOp()
171 std::vector<ScopedAllocator::Field> fields; in ExecOp()
290 std::vector<ScopedAllocator::Field> fields; in ExecOp()
Dscoped_allocator_ops.cc78 std::vector<ScopedAllocator::Field> fields_;
/external/tensorflow/tensorflow/core/framework/
Dtensor.h615 friend class ScopedAllocator; // For access to buf_. variable
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc300 std::vector<ScopedAllocator::Field> sa_fields; in AnalyzeInputs()