Searched refs:ScopedAllocatorInstance (Results 1 – 6 of 6) sorted by relevance
25 class ScopedAllocatorInstance; variable65 friend class ScopedAllocatorInstance;87 class ScopedAllocatorInstance : public Allocator {89 explicit ScopedAllocatorInstance(ScopedAllocator* sa, int32 field_index);92 ~ScopedAllocatorInstance() { VLOG(1) << "~ScopedAllocatorInstance " << this; } in ~ScopedAllocatorInstance()
136 ScopedAllocatorInstance::ScopedAllocatorInstance(ScopedAllocator* sa, in ScopedAllocatorInstance() function in tensorflow::ScopedAllocatorInstance147 void ScopedAllocatorInstance::DropFromTable() { in DropFromTable()166 void* ScopedAllocatorInstance::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw()187 void ScopedAllocatorInstance::DeallocateRaw(void* p) { in DeallocateRaw()206 string ScopedAllocatorInstance::Name() { in Name()
39 ScopedAllocatorInstance* GetInstance(int32 scope_id);59 ScopedAllocatorInstance* instance;61 SAField(int32 fi, ScopedAllocatorInstance* sai) in SAField()
92 std::vector<ScopedAllocatorInstance*> sa_instances_;156 ScopedAllocatorInstance* inst0 = sac->GetInstance(scope_id_ + 1); in TEST_F()162 ScopedAllocatorInstance* inst1 = sac->GetInstance(scope_id_ + 2); in TEST_F()166 ScopedAllocatorInstance* inst2 = sac->GetInstance(scope_id_ + 3); in TEST_F()
54 i, new ScopedAllocatorInstance(sa, i)); in AddScopedAllocator()73 ScopedAllocatorInstance* ScopedAllocatorContainer::GetInstance(int32 scope_id) { in GetInstance()
107 std::vector<ScopedAllocatorInstance*> sa_instances(fields_shapes.size(), in PrepOp()