Home
last modified time | relevance | path

Searched refs:ConvolutionDimensionNumbersToString (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dstream_executor_util.cc98 ConvolutionDimensionNumbersToString(dnums)); in StreamExecutorConvLayoutsToXlaLayouts()
129 ConvolutionDimensionNumbersToString(dnums)); in StreamExecutorConvLayoutsToXlaLayouts()
160 ConvolutionDimensionNumbersToString(dnums)); in StreamExecutorConvLayoutsToXlaLayouts()
213 ConvolutionDimensionNumbersToString(dnums), vect_size); in XlaConvShapesToStreamExecutorLayouts()
220 ConvolutionDimensionNumbersToString(dnums)); in XlaConvShapesToStreamExecutorLayouts()
237 ConvolutionDimensionNumbersToString(dnums), vect_size); in XlaConvShapesToStreamExecutorLayouts()
244 ConvolutionDimensionNumbersToString(dnums)); in XlaConvShapesToStreamExecutorLayouts()
261 ConvolutionDimensionNumbersToString(dnums), vect_size); in XlaConvShapesToStreamExecutorLayouts()
268 ConvolutionDimensionNumbersToString(dnums)); in XlaConvShapesToStreamExecutorLayouts()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instruction.h2417 std::string ConvolutionDimensionNumbersToString(
H A Dhlo_instructions.cc2480 extra.push_back(StrCat("dim_labels=", ConvolutionDimensionNumbersToString( in ExtraAttributesToStringImpl()
2760 ConvolutionDimensionNumbersToString(*convolution_dimension_numbers_))); in ExtraAttributesToStringImpl()
H A Dhlo_instruction.cc4127 std::string ConvolutionDimensionNumbersToString( in ConvolutionDimensionNumbersToString() function