| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | pool_allocator_test.cc | 40 pool.DeallocateRaw(nullptr); // Should not crash. in TEST()
72 pool.DeallocateRaw(p0); in TEST()
73 pool.DeallocateRaw(p4); in TEST()
74 pool.DeallocateRaw(p12); in TEST()
113 pool.DeallocateRaw(p); in TEST()
125 pool.DeallocateRaw(p); in TEST()
170 pool.DeallocateRaw(p1_16); in TEST()
177 pool.DeallocateRaw(p2_16); // Put it back. in TEST()
189 pool.DeallocateRaw(p3_4); // Put it back. Pool is now full. in TEST()
201 pool.DeallocateRaw(p4_2); in TEST()
[all …]
|
| D | gpu_bfc_allocator_test.cc | 129 a.DeallocateRaw(ptrs[i]); in TEST_P()
154 a.DeallocateRaw(initial_ptrs[i]); in TEST_P()
188 a.DeallocateRaw(existing_ptrs[i]); in TEST_P()
197 a.DeallocateRaw(first_ptr); in TEST_P()
209 a.DeallocateRaw(t1); in TEST_P()
210 a.DeallocateRaw(t2); in TEST_P()
211 a.DeallocateRaw(t3); in TEST_P()
212 a.DeallocateRaw(t4); in TEST_P()
224 a.DeallocateRaw(first_ptr_after); in TEST_P()
243 a.DeallocateRaw(t1); in TEST_P()
[all …]
|
| D | gpu_debug_allocator_test.cc | 67 a.DeallocateRaw(gpu_array); in TEST()
101 a.DeallocateRaw(gpu_array); in TEST()
137 a.DeallocateRaw(gpu_array); in TEST()
174 a.DeallocateRaw(gpu_array); in TEST()
219 a.DeallocateRaw(gpu_array); in TEST()
252 a.DeallocateRaw(t1); in TEST()
|
| D | gpu_debug_allocator.cc | 106 void GPUDebugAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::GPUDebugAllocator
115 base_allocator_->DeallocateRaw(ptr); in DeallocateRaw()
186 void GPUNanResetAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::GPUNanResetAllocator
202 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_managed_allocator.h | 31 void DeallocateRaw(void* ptr) override;
|
| D | gpu_allocator_retry_test.cc | 51 void DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::__anon55289c970111::FakeAllocator
144 alloc_->DeallocateRaw(ptr); in LaunchConsumerThreads()
|
| D | gpu_cudamalloc_allocator.h | 39 void DeallocateRaw(void* ptr) override;
|
| D | gpu_managed_allocator.cc | 46 void GpuManagedAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::GpuManagedAllocator
|
| 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 | cpu_allocator_impl.cc | 107 void DeallocateRaw(void* ptr) override { in DeallocateRaw() function in tensorflow::__anon435a67570111::CPUAllocator
166 cpu_allocator_->DeallocateRaw(ptr); in Free()
|
| D | allocator.h | 210 virtual void DeallocateRaw(void* ptr) = 0;
315 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 | 149 a->DeallocateRaw(ptrs[i]); in TEST()
236 a->DeallocateRaw(p); in BM_Allocation()
|
| /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()
213 EXPECT_DEATH(sa_instances_[0]->DeallocateRaw(ptr0 + 8), ""); in TEST_F()
223 sa_instances_[0]->DeallocateRaw(ptr0); in TEST_F()
224 sa_instances_[1]->DeallocateRaw(ptr1); in TEST_F()
|
| D | mkl_cpu_allocator.h | 72 void DeallocateRaw(void* ptr) override { in DeallocateRaw() function
240 inline void DeallocateRaw(void* ptr) override { in DeallocateRaw() function
244 small_size_allocator_->DeallocateRaw(ptr); in DeallocateRaw()
248 large_size_allocator_->DeallocateRaw(ptr); in DeallocateRaw()
286 cpu_allocator()->DeallocateRaw(ptr); in FreeHook()
|
| D | scoped_allocator.cc | 124 void ScopedAllocator::DeallocateRaw(void* p) { in DeallocateRaw() function in tensorflow::ScopedAllocator
212 void ScopedAllocatorInstance::DeallocateRaw(void* p) { in DeallocateRaw() function in tensorflow::ScopedAllocatorInstance
213 scoped_allocator_->DeallocateRaw(p); in DeallocateRaw()
|
| D | scoped_allocator.h | 70 void DeallocateRaw(void* p) TF_LOCKS_EXCLUDED(mu_);
108 void DeallocateRaw(void* p) TF_LOCKS_EXCLUDED(mu_) override;
|
| D | process_state.h | 132 void DeallocateRaw(void* p) override { in DeallocateRaw() function
136 a_->DeallocateRaw(p); in DeallocateRaw()
|
| D | bfc_allocator_test.cc | 78 bfc_allocator.DeallocateRaw(short_lived[deallocation_order[i]]); in BM_Allocator()
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_op.cc | 35 allocator_->DeallocateRaw(data()); in ~MlirTensorBuffer()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | tf_allocator_adapter.cc | 58 wrapped_->DeallocateRaw(mem.opaque()); in Deallocate()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | tf_framework_c_interface.cc | 73 GetAllocator(op_kernel_ctx)->DeallocateRaw(ptr); in _mlir_ciface_tf_dealloc()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_allocator.cc | 106 allocator_->DeallocateRaw(memory); in free()
|