Home
last modified time | relevance | path

Searched defs:AllocateSubBuffer (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.cc38 void *XlaInterpreterExecutor::AllocateSubBuffer(DeviceMemoryBase *parent, in AllocateSubBuffer() function in stream_executor::interpreter::XlaInterpreterExecutor
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.h846 DeviceMemory<T> StreamExecutor::AllocateSubBuffer(DeviceMemory<T> *parent, in AllocateSubBuffer() function
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc44 void *HostExecutor::AllocateSubBuffer(DeviceMemoryBase *parent, in AllocateSubBuffer() function in stream_executor::host::HostExecutor
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc429 void* GpuExecutor::AllocateSubBuffer(DeviceMemoryBase* mem, uint64 offset_bytes, in AllocateSubBuffer() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc548 void* GpuExecutor::AllocateSubBuffer(DeviceMemoryBase* mem, uint64 offset_bytes, in AllocateSubBuffer() function in stream_executor::gpu::GpuExecutor