Searched refs:ScopedAllocatorInstance (Results 1 – 6 of 6) sorted by relevance
25 class ScopedAllocatorInstance; variable66 friend class ScopedAllocatorInstance;88 class ScopedAllocatorInstance : public Allocator {90 explicit ScopedAllocatorInstance(ScopedAllocator* sa, int32 field_index);93 ~ScopedAllocatorInstance() override { in ~ScopedAllocatorInstance()
161 ScopedAllocatorInstance::ScopedAllocatorInstance(ScopedAllocator* sa, in ScopedAllocatorInstance() function in tensorflow::ScopedAllocatorInstance172 void ScopedAllocatorInstance::DropFromTable() { in DropFromTable()191 void* ScopedAllocatorInstance::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw()212 void ScopedAllocatorInstance::DeallocateRaw(void* p) { in DeallocateRaw()231 string ScopedAllocatorInstance::Name() { in Name()
40 ScopedAllocatorInstance* GetInstance(int32 scope_id);60 ScopedAllocatorInstance* instance;62 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()
108 std::vector<ScopedAllocatorInstance*> sa_instances(fields_shapes.size(), in PrepOp()