Home
last modified time | relevance | path

Searched defs:HostMemoryAllocate (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/
H A Dhost_gpu_executor.h68 void* HostMemoryAllocate(uint64_t size) override { return new char[size]; } in HostMemoryAllocate() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dgpu_executor.h159 void* HostMemoryAllocate(uint64_t size) override { in HostMemoryAllocate() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executor.h205 void* HostMemoryAllocate(uint64_t size) override { in HostMemoryAllocate() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dexecutor.h74 void *HostMemoryAllocate(uint64_t size) override { return new char[size]; } in HostMemoryAllocate() function
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/
H A Dstream_executor_test_util.cc27 void* HostMemoryAllocate(const SP_Device* const device, uint64_t size) { in HostMemoryAllocate() function
H A Dstream_executor.cc255 void* HostMemoryAllocate(uint64 size) override { in HostMemoryAllocate() function in stream_executor::__anon702966b10111::CStreamExecutor
H A Dstream_executor_test.cc190 TEST_F(StreamExecutorTest, HostMemoryAllocate) { in TEST_F() argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream_executor_pimpl.cc552 void* StreamExecutor::HostMemoryAllocate(uint64_t size) { in HostMemoryAllocate() function in stream_executor::StreamExecutor