Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dstream_executor_util.cc67 StreamExecutorConvLayoutsToXlaLayouts(const ConvolutionDimensionNumbers& dnums, in StreamExecutorConvLayoutsToXlaLayouts() function
178 StreamExecutorConvLayoutsToXlaLayouts(dnums, DataLayout::kBatchDepthYX, in XlaConvShapesToStreamExecutorLayouts()
186 StreamExecutorConvLayoutsToXlaLayouts(dnums, DataLayout::kBatchDepthYX4, in XlaConvShapesToStreamExecutorLayouts()
193 StreamExecutorConvLayoutsToXlaLayouts(dnums, DataLayout::kBatchYXDepth, in XlaConvShapesToStreamExecutorLayouts()
H A Dstream_executor_util.h40 StreamExecutorConvLayoutsToXlaLayouts(const ConvolutionDimensionNumbers& dnums,
H A Dgpu_layout_assignment.cc186 StreamExecutorConvLayoutsToXlaLayouts( in AddBackendConstraintsToDnnConvCustomCall()