Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.cc120 DeviceMemoryBase /*input_data*/, in GetConvolveRunners() argument
H A Dkernel.h87 class DeviceMemoryBase; variable
H A Dstream.h60 class DeviceMemoryBase; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executor.h51 using DeviceMemoryBase = ::stream_executor::DeviceMemoryBase; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc4714 DeviceMemoryBase /*input_data*/, in GetConvolveRunners() argument