| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.cc | 85 DeviceMemory<uint8> lhs_redzone = stream_->parent()->GetSubBuffer( in AllocateBytes()
88 DeviceMemory<uint8> data_chunk = stream_->parent()->GetSubBuffer( in AllocateBytes()
96 DeviceMemory<uint8> rhs_redzone_slop = stream_->parent()->GetSubBuffer( in AllocateBytes()
99 DeviceMemory<uint8> rhs_redzone_nonslop = stream_->parent()->GetSubBuffer( in AllocateBytes()
263 executor->GetSubBuffer(&buffer_uint8, 0, in CheckRedzonesForBuffer()
266 executor->GetSubBuffer(&buffer_uint8, redzone_size, in CheckRedzonesForBuffer()
269 executor->GetSubBuffer(&buffer_uint8, redzone_size + user_allocation_size, in CheckRedzonesForBuffer()
|
| D | gpu_executor.h | 91 void* GetSubBuffer(DeviceMemoryBase* mem, uint64 offset_bytes,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 165 DeviceMemory<T> GetSubBuffer(DeviceMemory<T> *parent, uint64 element_offset,
865 DeviceMemory<T> StreamExecutor::GetSubBuffer(DeviceMemory<T> *parent, in GetSubBuffer() function
875 void *opaque = implementation_->GetSubBuffer( in GetSubBuffer()
|
| D | stream_executor_internal.h | 202 virtual void *GetSubBuffer(DeviceMemoryBase *parent, uint64 offset,
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 41 void *XlaInterpreterExecutor::GetSubBuffer(DeviceMemoryBase *parent, in GetSubBuffer() function in stream_executor::interpreter::XlaInterpreterExecutor
|
| D | executor.h | 71 void *GetSubBuffer(DeviceMemoryBase *parent, uint64 offset_bytes,
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 64 void *GetSubBuffer(DeviceMemoryBase *parent, uint64 offset_bytes,
|
| 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/tpu/ |
| D | tpu_executor.h | 181 void* GetSubBuffer(DeviceMemoryBase* parent, uint64 offset, in GetSubBuffer() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transfer_manager.cc | 234 stream->parent()->GetSubBuffer(&buffer_8, offset, metadata_size); in ReadDynamicShapes()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 169 stream->parent()->GetSubBuffer(buffer, offset, metadata_size); in UpdateMetadata()
|
| /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::__anon00950a620111::CStreamExecutor
|
| /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
|