Home
last modified time | relevance | path

Searched defs:device_memory_base (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/
H A Dstream_executor.cc179 SP_DeviceMemoryBase device_memory_base{SP_DEVICE_MEMORY_BASE_STRUCT_SIZE}; in DeviceMemoryBaseToC() local
251 SP_DeviceMemoryBase device_memory_base = DeviceMemoryBaseToC(mem); in Deallocate() local
328 SP_DeviceMemoryBase device_memory_base = DeviceMemoryBaseToC(gpu_dst); in SynchronousMemcpy() local
337 SP_DeviceMemoryBase device_memory_base = DeviceMemoryBaseToC(&gpu_src); in SynchronousMemcpy() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executable_interface.cc136 se::DeviceMemoryBase device_memory_base = owning->Release(); in AllocateOutputMemoryWithInputReuse() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dexecutable_base.cc202 se::DeviceMemoryBase device_memory_base = owning->Release(); in AllocateOutputMemoryWithInputReuse() local