Searched defs:dim_nums (Results 1 – 6 of 6) sorted by relevance
41 DotDimensionNumbers dim_nums; member
60 DotDimensionNumbers dim_nums; member411 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in EmitNaiveLlvmIrGemm() local689 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetMatMultDims() local
105 ConvolutionDimensionNumbers dim_nums = in XLA_TEST_F() local
222 const DotDimensionNumbers& dim_nums = instruction->dot_dimension_numbers(); in MatchAndExplain() local
194 const DotDimensionNumbers& dim_nums = in AddBackendConstraints() local
323 DotDimensionNumbers dim_nums = GetDimensionNumbers(*hlo_instruction()); in ExecuteOnStream() local