| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | pool_allocator_test.cc | 39 pool.DeallocateRaw(nullptr); // Should not crash. in TEST()
70 pool.DeallocateRaw(p0); in TEST()
71 pool.DeallocateRaw(p4); in TEST()
72 pool.DeallocateRaw(p12); in TEST()
110 pool.DeallocateRaw(p); in TEST()
122 pool.DeallocateRaw(p); in TEST()
166 pool.DeallocateRaw(p1_16); in TEST()
173 pool.DeallocateRaw(p2_16); // Put it back. in TEST()
185 pool.DeallocateRaw(p3_4); // Put it back. Pool is now full. in TEST()
197 pool.DeallocateRaw(p4_2); in TEST()
[all …]
|
| D | gpu_bfc_allocator_test.cc | 80 a.DeallocateRaw(ptrs[i]); in TEST()
109 a.DeallocateRaw(initial_ptrs[i]); in TEST()
143 a.DeallocateRaw(existing_ptrs[i]); in TEST()
156 a.DeallocateRaw(first_ptr); in TEST()
168 a.DeallocateRaw(t1); in TEST()
169 a.DeallocateRaw(t2); in TEST()
170 a.DeallocateRaw(t3); in TEST()
171 a.DeallocateRaw(t4); in TEST()
183 a.DeallocateRaw(first_ptr_after); in TEST()
214 a.DeallocateRaw(t1); in TEST()
[all …]
|
| D | gpu_debug_allocator_test.cc | 57 a.DeallocateRaw(gpu_array); in TEST()
91 a.DeallocateRaw(gpu_array); in TEST()
127 a.DeallocateRaw(gpu_array); in TEST()
166 a.DeallocateRaw(gpu_array); in TEST()
213 a.DeallocateRaw(gpu_array); in TEST()
246 a.DeallocateRaw(t1); in TEST()
|
| D | gpu_debug_allocator.cc | 105 void GPUDebugAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::GPUDebugAllocator
114 base_allocator_->DeallocateRaw(ptr); in DeallocateRaw()
184 void GPUNanResetAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::GPUNanResetAllocator
200 base_allocator_->DeallocateRaw(ptr); in DeallocateRaw()
|
| D | gpu_debug_allocator.h | 41 void DeallocateRaw(void* ptr) override;
71 void DeallocateRaw(void* ptr) override;
|
| D | gpu_allocator_retry_test.cc | 51 void DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::__anon9dc664200111::FakeAllocator
144 alloc_->DeallocateRaw(ptr); in LaunchConsumerThreads()
|
| D | gpu_managed_allocator.h | 31 void DeallocateRaw(void* ptr) override;
|
| D | gpu_managed_allocator.cc | 37 void GpuManagedAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::GpuManagedAllocator
|
| D | gpu_cudamalloc_allocator.h | 39 void DeallocateRaw(void* ptr) override;
|
| D | gpu_cudamalloc_allocator.cc | 54 void GPUcudaMallocAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::GPUcudaMallocAllocator
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tracking_allocator_test.cc | 35 void DeallocateRaw(void* ptr) override { in DeallocateRaw() function in tensorflow::TestableSizeTrackingAllocator
59 void DeallocateRaw(void* ptr) override {} in DeallocateRaw() function in tensorflow::NoMemoryAllocator
75 ta->DeallocateRaw(p1); in TEST()
84 ta->DeallocateRaw(p2); in TEST()
96 ta->DeallocateRaw(p1); in TEST()
108 ta->DeallocateRaw(p2); in TEST()
124 ta->DeallocateRaw(p1); in TEST()
133 ta->DeallocateRaw(p2); in TEST()
168 ta->DeallocateRaw(nullptr); in TEST()
|
| D | allocator.h | 110 virtual void DeallocateRaw(void* ptr) = 0;
143 DeallocateRaw(ptr); in Deallocate()
309 void DeallocateRaw(void* ptr) override { wrapped_->DeallocateRaw(ptr); } in DeallocateRaw() function
|
| D | tracking_allocator.cc | 75 void TrackingAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::TrackingAllocator
106 allocator->DeallocateRaw(ptr); in DeallocateRaw()
|
| D | allocator_test.cc | 102 a->DeallocateRaw(ptrs[i]); in TEST()
185 a->DeallocateRaw(p); in BM_Allocation()
|
| D | allocator.cc | 143 void DeallocateRaw(void* ptr) override { in DeallocateRaw() function in tensorflow::__anonab749f280111::CPUAllocator
200 cpu_allocator_->DeallocateRaw(ptr); in Free()
|
| D | tracking_allocator.h | 64 void DeallocateRaw(void* ptr) override;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator_mgr_test.cc | 143 sa_instances_[0]->DeallocateRaw(ptr0); in TEST_F()
144 sa_instances_[1]->DeallocateRaw(ptr1); in TEST_F()
175 inst0->DeallocateRaw(ptr0); in TEST_F()
176 inst1->DeallocateRaw(ptr1); in TEST_F()
177 inst2->DeallocateRaw(ptr2); in TEST_F()
212 EXPECT_DEATH(sa_instances_[0]->DeallocateRaw(ptr0 + 8), ""); in TEST_F()
222 sa_instances_[0]->DeallocateRaw(ptr0); in TEST_F()
223 sa_instances_[1]->DeallocateRaw(ptr1); in TEST_F()
|
| D | mkl_cpu_allocator.h | 72 void DeallocateRaw(void* ptr) override { in DeallocateRaw() function
239 inline void DeallocateRaw(void* ptr) override { in DeallocateRaw() function
243 small_size_allocator_->DeallocateRaw(ptr); in DeallocateRaw()
247 large_size_allocator_->DeallocateRaw(ptr); in DeallocateRaw()
284 cpu_allocator()->DeallocateRaw(ptr); in FreeHook()
|
| D | scoped_allocator.cc | 99 void ScopedAllocator::DeallocateRaw(void* p) { in DeallocateRaw() function in tensorflow::ScopedAllocator
187 void ScopedAllocatorInstance::DeallocateRaw(void* p) { in DeallocateRaw() function in tensorflow::ScopedAllocatorInstance
188 scoped_allocator_->DeallocateRaw(p); in DeallocateRaw()
|
| D | scoped_allocator.h | 69 void DeallocateRaw(void* p) LOCKS_EXCLUDED(mu_);
105 void DeallocateRaw(void* p) LOCKS_EXCLUDED(mu_) override;
|
| D | process_state.h | 124 void DeallocateRaw(void* p) override { in DeallocateRaw() function
128 a_->DeallocateRaw(p); in DeallocateRaw()
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_allocator.h | 36 void DeallocateRaw(void* ptr) override;
|
| D | sycl_allocator.cc | 59 void SYCLAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::SYCLAllocator
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | immutable_constant_op.cc | 52 void DeallocateRaw(void* ptr) override { in DeallocateRaw() function in tensorflow::__anon173c32940111::MemmappedTensorAllocator
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_allocator.cc | 111 allocator_->DeallocateRaw(memory); in free()
|