Searched defs:dim_nums (Results 1 – 9 of 9) sorted by relevance
39 const ConvolutionDimensionNumbers& dim_nums = in InstructionMatchesPattern() local59 ConvolutionDimensionNumbers dim_nums = in ExpandInstruction() local
182 ScatterDimensionNumbers dim_nums = in ExpandInstruction() local
232 const DotDimensionNumbers& dim_nums = instruction->dot_dimension_numbers(); in MatchAndExplain() local
41 DotDimensionNumbers dim_nums; member
72 DotDimensionNumbers dim_nums; member618 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in EmitNaiveLlvmIrGemm() local1008 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetMatMultDims() local1038 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetBatchMatMultDims() local
51 DotDimensionNumbers *dim_nums = config.mutable_dot_dimension_numbers(); in HandleCustomCall() local
102 ConvolutionDimensionNumbers dim_nums = in XLA_TEST_F() local
221 const mlir::mhlo::DotDimensionNumbers dim_nums = in GemmOpConversionRewrite() local
534 const auto& dim_nums = conv->convolution_dimension_numbers(); in EmitConvolutionForwardAsMlir() local