Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_host_allocator.h42 ptr = stream_exec_->HostMemoryAllocate(num_bytes); in Alloc()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h76 void *HostMemoryAllocate(uint64 size) override { return new char[size]; } in HostMemoryAllocate() function
/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/stream_executor/
Dstream_executor_internal.h211 virtual void *HostMemoryAllocate(uint64 size) = 0;
Dstream_executor_pimpl.cc562 void *StreamExecutor::HostMemoryAllocate(uint64 size) { in HostMemoryAllocate() function in stream_executor::StreamExecutor
563 void *buffer = implementation_->HostMemoryAllocate(size); in HostMemoryAllocate()
Dstream_executor_pimpl.h226 void *HostMemoryAllocate(uint64 bytes);
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h108 void* HostMemoryAllocate(uint64 size) override { in HostMemoryAllocate() function