Home
last modified time | relevance | path

Searched refs:GetMutableDeviceAddress (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dbuffer_allocations.h62 se::DeviceMemoryBase& GetMutableDeviceAddress(
H A Dbuffer_allocations.cc65 se::DeviceMemoryBase& BufferAllocations::GetMutableDeviceAddress( in GetMutableDeviceAddress() function in xla::gpu::BufferAllocations
H A Dgpu_executable.cc806 buffer_allocations.GetMutableDeviceAddress( in ExecuteAsyncOnStreamImpl()