Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.h145 using BufferAllocToDeviceMemoryMap = variable
151 StatusOr<const BufferAllocToDeviceMemoryMap*> ResolveConstantGlobals(
161 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals,
167 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals,
213 std::map<stream_executor::StreamExecutor*, BufferAllocToDeviceMemoryMap>
Dgpu_executable.cc256 StatusOr<const GpuExecutable::BufferAllocToDeviceMemoryMap*>
304 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals, in BufferForAllocation()
376 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals, in GenerateBufferAllocations()
426 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals; in ExecuteAsyncOnStreamImpl()