Searched refs:mlir_dims (Results 1 – 2 of 2) sorted by relevance
668 auto mlir_dims = mhlo::DotDimensionNumbers::get( in EmitGemm() local674 op.dot_dimension_numbersAttr(mlir_dims); in EmitGemm()
1239 auto mlir_dims = op.dot_dimension_numbers(); in EmitGemmThunkFromMlir() local1245 fill_dims(mlir_dims.lhs_batching_dimensions(), in EmitGemmThunkFromMlir()1247 fill_dims(mlir_dims.rhs_batching_dimensions(), in EmitGemmThunkFromMlir()1249 fill_dims(mlir_dims.lhs_contracting_dimensions(), in EmitGemmThunkFromMlir()1251 fill_dims(mlir_dims.rhs_contracting_dimensions(), in EmitGemmThunkFromMlir()