| /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 | 83 a.DeallocateRaw(ptrs[i]); in TEST()
112 a.DeallocateRaw(initial_ptrs[i]); in TEST()
146 a.DeallocateRaw(existing_ptrs[i]); in TEST()
159 a.DeallocateRaw(first_ptr); in TEST()
171 a.DeallocateRaw(t1); in TEST()
172 a.DeallocateRaw(t2); in TEST()
173 a.DeallocateRaw(t3); in TEST()
174 a.DeallocateRaw(t4); in TEST()
186 a.DeallocateRaw(first_ptr_after); in TEST()
217 a.DeallocateRaw(t1); in TEST()
[all …]
|
| D | gpu_debug_allocator_test.cc | 60 a.DeallocateRaw(gpu_array); in TEST()
95 a.DeallocateRaw(gpu_array); in TEST()
132 a.DeallocateRaw(gpu_array); in TEST()
171 a.DeallocateRaw(gpu_array); in TEST()
218 a.DeallocateRaw(gpu_array); in TEST()
251 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::__anon3016a1e20111::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 | 151 virtual void DeallocateRaw(void* ptr) = 0;
256 void DeallocateRaw(void* ptr) override { wrapped_->DeallocateRaw(ptr); } in DeallocateRaw() function
|
| D | cpu_allocator_impl.cc | 108 void DeallocateRaw(void* ptr) override { in DeallocateRaw() function in tensorflow::__anone20f20020111::CPUAllocator
165 cpu_allocator_->DeallocateRaw(ptr); in Free()
|
| 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()
234 a->DeallocateRaw(p); in BM_Allocation()
|
| D | tracking_allocator.h | 63 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()
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
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()
285 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) LOCKS_EXCLUDED(mu_);
108 void DeallocateRaw(void* p) LOCKS_EXCLUDED(mu_) override;
|
| D | process_state.h | 125 void DeallocateRaw(void* p) override { in DeallocateRaw() function
129 a_->DeallocateRaw(p); in DeallocateRaw()
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_allocator.h | 36 void DeallocateRaw(void* ptr) override;
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | tf_allocator_adapter.cc | 58 wrapped_->DeallocateRaw(mem.opaque()); in Deallocate()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | immutable_constant_op.cc | 52 void DeallocateRaw(void* ptr) override { in DeallocateRaw() function in tensorflow::__anon4fc2a4d60111::MemmappedTensorAllocator
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_allocator.cc | 101 allocator_->DeallocateRaw(memory); in free()
|