Searched defs:dim_nums (Results 1 – 8 of 8) sorted by relevance
39 const ConvolutionDimensionNumbers& dim_nums = in InstructionMatchesPattern() local59 ConvolutionDimensionNumbers dim_nums = in ExpandInstruction() local
226 const DotDimensionNumbers& dim_nums = instruction->dot_dimension_numbers(); in MatchAndExplain() local
41 DotDimensionNumbers dim_nums; member
72 DotDimensionNumbers dim_nums; member554 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in EmitNaiveLlvmIrGemm() local855 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetMatMultDims() local
105 ConvolutionDimensionNumbers dim_nums = in XLA_TEST_F() local
218 DotDimensionNumbers dim_nums = instruction->dot_dimension_numbers(); in AddBackendConstraints() local
186 const DotDimensionNumbers &dim_nums = backend_config.dot_dimension_numbers(); in RunGemm() local
534 const auto& dim_nums = conv->convolution_dimension_numbers(); in EmitConvolutionForwardAsMlir() local