Home
last modified time | relevance | path

Searched defs:UnifiedMemoryAllocate (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h96 void* UnifiedMemoryAllocate(uint64 size) override { in UnifiedMemoryAllocate() function
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h206 virtual void *UnifiedMemoryAllocate(uint64 size) { return nullptr; } in UnifiedMemoryAllocate() function
Dstream_executor_pimpl.cc548 void *StreamExecutor::UnifiedMemoryAllocate(uint64 bytes) { in UnifiedMemoryAllocate() function in stream_executor::StreamExecutor
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_driver.cc684 /* static */ void* GpuDriver::UnifiedMemoryAllocate(GpuContext* context, in UnifiedMemoryAllocate() function in stream_executor::gpu::GpuDriver
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc860 /* static */ void* GpuDriver::UnifiedMemoryAllocate(GpuContext* context, in UnifiedMemoryAllocate() function in stream_executor::gpu::GpuDriver