Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dgpu_executor.h147 void* UnifiedMemoryAllocate(uint64_t size) override { in UnifiedMemoryAllocate() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream_executor_internal.h206 virtual void* UnifiedMemoryAllocate(uint64_t size) { return nullptr; } in UnifiedMemoryAllocate() function
H A Dstream_executor_pimpl.cc538 void* StreamExecutor::UnifiedMemoryAllocate(uint64_t bytes) { in UnifiedMemoryAllocate() function in stream_executor::StreamExecutor
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/
H A Dstream_executor.cc266 void* UnifiedMemoryAllocate(uint64 size) override { in UnifiedMemoryAllocate() function in stream_executor::__anon702966b10111::CStreamExecutor
H A Dstream_executor_test.cc211 TEST_F(StreamExecutorTest, UnifiedMemoryAllocate) { in TEST_F() argument
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_driver.cc640 /* static */ void* GpuDriver::UnifiedMemoryAllocate(GpuContext* context, in UnifiedMemoryAllocate() function in stream_executor::gpu::GpuDriver
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_driver.cc758 /* static */ void* GpuDriver::UnifiedMemoryAllocate(GpuContext* context, in UnifiedMemoryAllocate() function in stream_executor::gpu::GpuDriver