Home
last modified time | relevance | path

Searched refs:BufferAllocToDeviceMemoryMap (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.h219 using BufferAllocToDeviceMemoryMap = variable
232 StatusOr<const BufferAllocToDeviceMemoryMap*> ResolveConstantGlobals(
242 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals,
247 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals,
295 std::map<stream_executor::StreamExecutor*, BufferAllocToDeviceMemoryMap>
H A Dgpu_executable.cc385 StatusOr<const GpuExecutable::BufferAllocToDeviceMemoryMap*>
459 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals, in BufferForAllocation()
534 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals, in GenerateBufferAllocations()
700 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals; in ExecuteAsyncOnStreamImpl()