Home
last modified time | relevance | path

Searched refs:DeallocateRaw (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dpool_allocator_test.cc39 pool.DeallocateRaw(nullptr); // Should not crash. in TEST()
69 pool.DeallocateRaw(p0); in TEST()
70 pool.DeallocateRaw(p4); in TEST()
71 pool.DeallocateRaw(p12); in TEST()
107 pool.DeallocateRaw(p); in TEST()
119 pool.DeallocateRaw(p); in TEST()
141 pool.DeallocateRaw(p1_16); in TEST()
148 pool.DeallocateRaw(p2_16); // Put it back. in TEST()
157 pool.DeallocateRaw(p3_4); // Put it back. Pool is now full. in TEST()
166 pool.DeallocateRaw(p4_2); in TEST()
[all …]
Dgpu_bfc_allocator_test.cc72 a.DeallocateRaw(ptrs[i]); in TEST()
97 a.DeallocateRaw(initial_ptrs[i]); in TEST()
131 a.DeallocateRaw(existing_ptrs[i]); in TEST()
140 a.DeallocateRaw(first_ptr); in TEST()
152 a.DeallocateRaw(t1); in TEST()
153 a.DeallocateRaw(t2); in TEST()
154 a.DeallocateRaw(t3); in TEST()
155 a.DeallocateRaw(t4); in TEST()
167 a.DeallocateRaw(first_ptr_after); in TEST()
186 a.DeallocateRaw(t1); in TEST()
[all …]
Dgpu_debug_allocator_test.cc53 a.DeallocateRaw(gpu_array); in TEST()
84 a.DeallocateRaw(gpu_array); in TEST()
117 a.DeallocateRaw(gpu_array); in TEST()
152 a.DeallocateRaw(gpu_array); in TEST()
195 a.DeallocateRaw(gpu_array); in TEST()
222 a.DeallocateRaw(t1); in TEST()
Dgpu_debug_allocator.cc104 void GPUDebugAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::GPUDebugAllocator
111 base_allocator_->DeallocateRaw(ptr); in DeallocateRaw()
184 void GPUNanResetAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::GPUNanResetAllocator
195 base_allocator_->DeallocateRaw(ptr); in DeallocateRaw()
Dgpu_debug_allocator.h41 void DeallocateRaw(void* ptr) override;
73 void DeallocateRaw(void* ptr) override;
Dprocess_state.h151 void DeallocateRaw(void* p) override { in DeallocateRaw() function
155 a_->DeallocateRaw(p); in DeallocateRaw()
Dgpu_allocator_retry_test.cc51 void DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::__anonb320a5940111::FakeAllocator
144 alloc_->DeallocateRaw(ptr); in LaunchConsumerThreads()
Dgpu_managed_allocator.h31 void DeallocateRaw(void* ptr) override;
Dgpu_managed_allocator.cc33 void GpuManagedAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::GpuManagedAllocator
Dgpu_cudamalloc_allocator.h39 void DeallocateRaw(void* ptr) override;
Dgpu_cudamalloc_allocator.cc53 void GPUcudaMallocAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::GPUcudaMallocAllocator
/external/tensorflow/tensorflow/core/framework/
Dtracking_allocator_test.cc35 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()
Dallocator.h102 virtual void DeallocateRaw(void* ptr) = 0;
135 DeallocateRaw(ptr); in Deallocate()
305 void DeallocateRaw(void* ptr) override { wrapped_->DeallocateRaw(ptr); } in DeallocateRaw() function
Dallocator_test.cc99 a->DeallocateRaw(ptrs[i]); in TEST()
182 a->DeallocateRaw(p); in BM_Allocation()
Dtracking_allocator.cc75 void TrackingAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::TrackingAllocator
106 allocator->DeallocateRaw(ptr); in DeallocateRaw()
Dtracking_allocator.h65 void DeallocateRaw(void* ptr) override;
Dallocator.cc94 void DeallocateRaw(void* ptr) override { in DeallocateRaw() function in tensorflow::CPUAllocator
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_cpu_allocator.h113 inline void DeallocateRaw(void* ptr) override { in DeallocateRaw() function
114 allocator_->DeallocateRaw(ptr); in DeallocateRaw()
139 cpu_allocator()->DeallocateRaw(ptr); in FreeHook()
Dvisitable_allocator.h63 void DeallocateRaw(void* ptr) override { in DeallocateRaw() function
64 TrackingAllocator::DeallocateRaw(ptr); in DeallocateRaw()
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_allocator.h36 void DeallocateRaw(void* ptr) override;
Dsycl_allocator.cc59 void SYCLAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::SYCLAllocator
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.h39 void DeallocateRaw(void* ptr) override;
Dxla_device_context.cc47 void XlaDeviceAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw() function in tensorflow::XlaDeviceAllocator
/external/tensorflow/tensorflow/core/kernels/
Dimmutable_constant_op.cc52 void DeallocateRaw(void* ptr) override { in DeallocateRaw() function in tensorflow::__anonb22639080111::MemmappedTensorAllocator
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compilation_device.cc52 void DeallocateRaw(void* ptr) override { in DeallocateRaw() function in tensorflow::XlaCompilationAllocator

12