Home
last modified time | relevance | path

Searched defs:UnifiedMemoryAllocate (Results 1 – 7 of 7) 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.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
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc294 void* UnifiedMemoryAllocate(uint64 size) override { in UnifiedMemoryAllocate() function in stream_executor::__anon902a2f7b0111::CStreamExecutor
Dstream_executor_test.cc387 TEST_F(StreamExecutorTest, UnifiedMemoryAllocate) { in TEST_F() argument
/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