(TF_MatMulOp $a, (TF_TransposeOp $b, (TF_SubOp (TF_RangeOp
             /*start=*/(TF_RankOp $b),
             /*limit=*/(TF_ConstOp TFi32<0>),
             /*delta=*/(TF_ConstOp TFi32<-1>)), (TF_ConstOp TFi32<1>))),
           $at, ConstBoolAttrTrue, $grad_a, $grad_b)>;

// Matmul with transpose on a to matmul with explicit transpose op and a not
// transposed.

  [(TF_OneHotOp:$labels $sparse_labels,
     (TF_ConstOp (GetDimSizeAsI32<1> $features, $src_op__0)),
     (TF_ConstOp (GetScalarOfType<1> $features)),
     (TF_ConstOp (GetScalarOfType<0> $features)),
     ConstantAttr<I64Attr, "1">),
   (TF_SelectV2Op:$zero_or_nan
     (TF_LogicalAndOp
       (TF_LessEqualOp
         (TF_ConstOp (GetScalarOfType<0> $sparse_labels)), $sparse_labels),

            (TF_BitwiseXorOp $input,
              (TF_ConstOp (GetRankedScalarAttr<"", 8, "", "-1">)))>;
def : Pat<(MHLO_NotOp TF_Int16Tensor:$input),
            (TF_BitwiseXorOp $input,
              (TF_ConstOp (GetRankedScalarAttr<"", 16, "", "-1">)))>;
def : Pat<(MHLO_NotOp TF_Int32Tensor:$input),
            (TF_BitwiseXorOp $input,
              (TF_ConstOp (GetRankedScalarAttr<"", 32, "", "-1">)))>;

     $l2_shrinkage, $lr_power, $use_locking, $multiply_linear_by_lr),
  [// grad_with_shrinkage = grad + 2 * l2_shrinkage * var
   (TF_ConstOp:$zero (GetScalarOfType<0> $lr)),
   (TF_ConstOp:$one (GetScalarOfType<1> $lr)),
   (TF_ConstOp:$two (GetScalarOfType<2> $lr)),
   (TF_AddV2Op:$grad_with_shrinkage
     $grad,
     (TF_MulOp
       $two,

//===----------------------------------------------------------------------===//

def MatrixDiagToV3 : Pat<
  (TF_MatrixDiagOp:$src $diag),
  (TF_MatrixDiagV3Op:$dest $diag, (TF_ConstOp (GetI32Attr<0>)),
    (TF_ConstOp (GetI32Attr<-1>)), (TF_ConstOp (GetI32Attr<-1>)),
    (TF_ConstOp (GetScalarOfType<0> $diag)), (GetStrAttr<"RIGHT_LEFT">)),
  [], [(CopyAttrs $src, $dest)]>;

//===----------------------------------------------------------------------===//

def : Pat<(TF_RiscAddOp $l, $r), (MHLO_AddOp $l, $r)>;

def : Pat<(TF_RiscDotOp $a, $b, $transpose_a, $transpose_b),
          (MHLO_DotOp
          (TF_TransposeOp $a, (TF_ConstOp (Get2DTransposePerm $transpose_a))),
          (TF_TransposeOp $b, (TF_ConstOp (Get2DTransposePerm $transpose_b))),
          /*precision_config=*/(NullArrayAttr))>;

//===----------------------------------------------------------------------===//

  : NativeCodeCall<"$_builder.create<ConstOp>($_loc, $0.getType(), $1)">;

def LegalizeTFConstToTFLConst: Pat<(TF_ConstOp:$res ElementsAttr:$value),
                                   (createConstOp $res, $value)>;

// Convert to std constant for statically shaped, non-opaque constants.
def ConvertTfConstToStdConst : Pat<
  (TF_ConstOp:$res DenseElementsAttr:$value),
  (Arith_ConstantOp $value),
  [(AnyStaticShapeTensor $res)], [], (addBenefit 10)>;