Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dstream_executor_util.h48 XlaConvShapesToStreamExecutorLayouts(const ConvolutionDimensionNumbers& dnums,
H A Dstream_executor_util.cc169 XlaConvShapesToStreamExecutorLayouts(const ConvolutionDimensionNumbers& dnums, in XlaConvShapesToStreamExecutorLayouts() function
H A Dgpu_conv_runner.cc390 XlaConvShapesToStreamExecutorLayouts( in GetGpuConvConfig()