Searched defs:GetSubBuffer (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 41 void *XlaInterpreterExecutor::GetSubBuffer(DeviceMemoryBase *parent, in GetSubBuffer() function in stream_executor::interpreter::XlaInterpreterExecutor
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executor.h | 181 void* GetSubBuffer(DeviceMemoryBase* parent, uint64 offset, in GetSubBuffer() function
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 865 DeviceMemory<T> StreamExecutor::GetSubBuffer(DeviceMemory<T> *parent, in GetSubBuffer() function
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 78 void *HostExecutor::GetSubBuffer(DeviceMemoryBase *parent, uint64 offset_bytes, in GetSubBuffer() function in stream_executor::host::HostExecutor
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 522 void* GpuExecutor::GetSubBuffer(DeviceMemoryBase* mem, uint64 offset_bytes, in GetSubBuffer() function in stream_executor::gpu::GpuExecutor
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 430 void* GpuExecutor::GetSubBuffer(DeviceMemoryBase* mem, uint64 offset_bytes, in GetSubBuffer() function in stream_executor::gpu::GpuExecutor
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.cc | 273 void* GetSubBuffer(DeviceMemoryBase* parent, uint64 offset, in GetSubBuffer() function in stream_executor::__anon902a2f7b0111::CStreamExecutor
|