Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_mem_allocator.h50 ptr = stream_exec_->UnifiedMemoryAllocate(num_bytes); in Alloc()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h96 void* UnifiedMemoryAllocate(uint64 size) override { in UnifiedMemoryAllocate() function
97 return GpuDriver::UnifiedMemoryAllocate(context_, size); in UnifiedMemoryAllocate()
Dgpu_driver.h113 static void* UnifiedMemoryAllocate(GpuContext* context, uint64 bytes);
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h209 virtual void *UnifiedMemoryAllocate(uint64 size) { return nullptr; } in UnifiedMemoryAllocate() function
Dstream_executor_pimpl.cc535 void *StreamExecutor::UnifiedMemoryAllocate(uint64 bytes) { in UnifiedMemoryAllocate() function in stream_executor::StreamExecutor
536 void *buffer = implementation_->UnifiedMemoryAllocate(bytes); in UnifiedMemoryAllocate()
Dstream_executor_pimpl.h204 void *UnifiedMemoryAllocate(uint64 bytes);
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor_test.cc387 TEST_F(StreamExecutorTest, UnifiedMemoryAllocate) { in TEST_F() argument
401 void* mem = executor->UnifiedMemoryAllocate(8); in TEST_F()
Dstream_executor.cc294 void* UnifiedMemoryAllocate(uint64 size) override { in UnifiedMemoryAllocate() function in stream_executor::__anon00950a620111::CStreamExecutor
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc632 /* static */ void* GpuDriver::UnifiedMemoryAllocate(GpuContext* context, in UnifiedMemoryAllocate() function in stream_executor::gpu::GpuDriver
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc814 /* static */ void* GpuDriver::UnifiedMemoryAllocate(GpuContext* context, in UnifiedMemoryAllocate() function in stream_executor::gpu::GpuDriver