Home
last modified time | relevance | path

Searched refs:UnifiedMemoryDeallocate (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_mem_allocator.h63 stream_exec_->UnifiedMemoryDeallocate(ptr); in Free()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h100 void UnifiedMemoryDeallocate(void* location) override { in UnifiedMemoryDeallocate() function
101 return GpuDriver::UnifiedMemoryDeallocate(context_, location); in UnifiedMemoryDeallocate()
Dgpu_driver.h119 static void UnifiedMemoryDeallocate(GpuContext* context, void* location);
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h213 virtual void UnifiedMemoryDeallocate(void *mem) {} in UnifiedMemoryDeallocate() function
Dstream_executor_pimpl.cc542 void StreamExecutor::UnifiedMemoryDeallocate(void *location) { in UnifiedMemoryDeallocate() function in stream_executor::StreamExecutor
546 return implementation_->UnifiedMemoryDeallocate(location); in UnifiedMemoryDeallocate()
Dstream_executor_pimpl.h208 void UnifiedMemoryDeallocate(void *location);
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc299 void UnifiedMemoryDeallocate(void* mem) override { in UnifiedMemoryDeallocate() function in stream_executor::__anon00950a620111::CStreamExecutor
Dstream_executor_test.cc405 executor->UnifiedMemoryDeallocate(mem); in TEST_F()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc641 /* static */ void GpuDriver::UnifiedMemoryDeallocate(GpuContext* context, in UnifiedMemoryDeallocate() function in stream_executor::gpu::GpuDriver
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc831 /* static */ void GpuDriver::UnifiedMemoryDeallocate(GpuContext* context, in UnifiedMemoryDeallocate() function in stream_executor::gpu::GpuDriver