Home
last modified time | relevance | path

Searched refs:HostMemoryAllocate (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dmemcpy_test.cc34 void* d_ptr = executor->HostMemoryAllocate(sizeof(int)); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_host_allocator.h44 ptr = stream_exec_->HostMemoryAllocate(num_bytes); in Alloc()
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h68 void *HostMemoryAllocate(uint64 size) override { return new char[size]; } in HostMemoryAllocate() function
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h75 void *HostMemoryAllocate(uint64 size) override { return new char[size]; } in HostMemoryAllocate() function
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.h200 void* HostMemoryAllocate(uint64 size) override { in HostMemoryAllocate() function
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h214 virtual void *HostMemoryAllocate(uint64 size) = 0;
Dstream_executor_pimpl.cc549 void *StreamExecutor::HostMemoryAllocate(uint64 size) { in HostMemoryAllocate() function in stream_executor::StreamExecutor
550 void *buffer = implementation_->HostMemoryAllocate(size); in HostMemoryAllocate()
Dstream_executor_pimpl.h214 void *HostMemoryAllocate(uint64 size);
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h108 void* HostMemoryAllocate(uint64 size) override { in HostMemoryAllocate() function
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor_test.cc366 TEST_F(StreamExecutorTest, HostMemoryAllocate) { in TEST_F() argument
379 void* mem = executor->HostMemoryAllocate(8); in TEST_F()
Dstream_executor.cc283 void* HostMemoryAllocate(uint64 size) override { in HostMemoryAllocate() function in stream_executor::__anon00950a620111::CStreamExecutor