Searched refs:AllocateSubBuffer (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 164 DeviceMemory<T> AllocateSubBuffer(DeviceMemory<T> *parent,
174 this, AllocateSubBuffer<T>(parent, element_offset, element_count)); in AllocateOwnedSubBuffer()
846 DeviceMemory<T> StreamExecutor::AllocateSubBuffer(DeviceMemory<T> *parent, in AllocateSubBuffer() function
856 void *opaque = implementation_->AllocateSubBuffer( in AllocateSubBuffer()
|
| D | stream_executor_internal.h | 199 virtual void *AllocateSubBuffer(DeviceMemoryBase *parent, uint64 offset,
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 38 void *XlaInterpreterExecutor::AllocateSubBuffer(DeviceMemoryBase *parent, in AllocateSubBuffer() function in stream_executor::interpreter::XlaInterpreterExecutor
|
| D | executor.h | 72 void *AllocateSubBuffer(DeviceMemoryBase *mem, uint64 offset_bytes,
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 64 void *AllocateSubBuffer(DeviceMemoryBase *mem, uint64 offset_bytes,
|
| D | host_gpu_executor.cc | 44 void *HostExecutor::AllocateSubBuffer(DeviceMemoryBase *parent, in AllocateSubBuffer() function in stream_executor::host::HostExecutor
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 91 void* AllocateSubBuffer(DeviceMemoryBase* mem, uint64 offset_bytes,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 429 void* GpuExecutor::AllocateSubBuffer(DeviceMemoryBase* mem, uint64 offset_bytes, in AllocateSubBuffer() function in stream_executor::gpu::GpuExecutor
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 548 void* GpuExecutor::AllocateSubBuffer(DeviceMemoryBase* mem, uint64 offset_bytes, in AllocateSubBuffer() function in stream_executor::gpu::GpuExecutor
|