Searched defs:GetSubBuffer (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| H A D | executor.cc | 41 void *XlaInterpreterExecutor::GetSubBuffer(DeviceMemoryBase *parent, in GetSubBuffer() function in stream_executor::interpreter::XlaInterpreterExecutor
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | tpu_executor.h | 186 void* GetSubBuffer(DeviceMemoryBase* parent, uint64_t offset, in GetSubBuffer() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream_executor_pimpl.h | 883 DeviceMemory<T> StreamExecutor::GetSubBuffer(DeviceMemory<T>* parent, in GetSubBuffer() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/ |
| H A D | host_gpu_executor.cc | 78 void* HostExecutor::GetSubBuffer(DeviceMemoryBase* parent, in GetSubBuffer() function in stream_executor::host::HostExecutor
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_gpu_executor.cc | 432 void* GpuExecutor::GetSubBuffer(DeviceMemoryBase* mem, uint64_t offset_bytes, in GetSubBuffer() function in stream_executor::gpu::GpuExecutor
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| H A D | stream_executor.cc | 245 void* GetSubBuffer(DeviceMemoryBase* parent, uint64 offset, in GetSubBuffer() function in stream_executor::__anon702966b10111::CStreamExecutor
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_gpu_executor.cc | 564 void* GpuExecutor::GetSubBuffer(DeviceMemoryBase* mem, uint64_t offset_bytes, in GetSubBuffer() function in stream_executor::gpu::GpuExecutor
|