Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h108 void* HostMemoryAllocate(uint64 size) override { 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/tpu/
Dtpu_executor.h200 void* HostMemoryAllocate(uint64 size) override { 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/c/experimental/stream_executor/
Dstream_executor.cc283 void* HostMemoryAllocate(uint64 size) override { in HostMemoryAllocate() function in stream_executor::__anon00950a620111::CStreamExecutor
Dstream_executor_test.cc366 TEST_F(StreamExecutorTest, HostMemoryAllocate) { in TEST_F() argument
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc549 void *StreamExecutor::HostMemoryAllocate(uint64 size) { in HostMemoryAllocate() function in stream_executor::StreamExecutor