Searched refs:dot_op (Results 1 – 3 of 3) sorted by relevance
543 auto dot_op = cast<mhlo::DotOp>(old_op); in ConvertDotOp() local544 auto lhs_rank = dot_op.lhs().getType().cast<ShapedType>().getRank(); in ConvertDotOp()552 return ConvertDot(rewriter, dot_op.lhs(), dot_op.rhs(), dot_dimension_numbers, in ConvertDotOp()553 dot_op.getResult().getType().cast<ShapedType>(), in ConvertDotOp()554 dot_op.getLoc()); in ConvertDotOp()
283 multiclass DotPats<SDPatternOperator dot_op,287 (dot_op (dot_inst.Pfl.Src0VT (VOP3PMods0 dot_inst.Pfl.Src0VT:$src0, i32:$src0_modifiers)),
1078 DotOperationType GetDotOperationType(mhlo::DotOp dot_op) { in GetDotOperationType() argument1080 dot_op.lhs().getType().cast<ShapedType>().getShape(); in GetDotOperationType()1082 dot_op.rhs().getType().cast<ShapedType>().getShape(); in GetDotOperationType()