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/compiler/xla/stream_executor/tpu/ |
| D | tpu_executor.h | 186 void* GetSubBuffer(DeviceMemoryBase* parent, uint64_t offset, in GetSubBuffer() function
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream_executor_pimpl.h | 883 DeviceMemory<T> StreamExecutor::GetSubBuffer(DeviceMemory<T>* parent, in GetSubBuffer() function
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| D | host_gpu_executor.cc | 78 void* HostExecutor::GetSubBuffer(DeviceMemoryBase* parent, in GetSubBuffer() function in stream_executor::host::HostExecutor
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.cc | 245 void* GetSubBuffer(DeviceMemoryBase* parent, uint64 offset, in GetSubBuffer() function in stream_executor::__anon7d2d81310111::CStreamExecutor
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 432 void* GpuExecutor::GetSubBuffer(DeviceMemoryBase* mem, uint64_t offset_bytes, in GetSubBuffer() function in stream_executor::gpu::GpuExecutor
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 564 void* GpuExecutor::GetSubBuffer(DeviceMemoryBase* mem, uint64_t offset_bytes, in GetSubBuffer() function in stream_executor::gpu::GpuExecutor
|