// incorrect placement in the quantization aware training.
def MatchHardSwishQuantized : Pat<
  (TFL_MulOp (TFL_DequantizeOp (TFL_QuantizeOp
    (TFL_MulOp
     $x, (TFL_DequantizeOp (TFL_QuantizeOp (TFL_AddOp
          $x,
          (Arith_ConstantOp ConstantAttr<RankedF32ElementsAttr<[]>, "3.0f">),
          TFL_AF_Relu6), $qattr2)),
     TFL_AF_None), $qattr1)),

// def UndoBroadcastFullyConnectedBiasAddWithQDQs : Pat<
//   (TFL_AddOp $lhs,
//     (TFL_DequantizeOp
//       (TFL_QuantizeOp
//         (Arith_ConstantOp:$const_op $bias),
//       $qparams)),
//   $act_fn),
//   (TFL_AddOp $lhs,
//     (TFL_DequantizeOp
//       (TFL_QuantizeOp
//         (Arith_ConstantOp:$const_op (FlattenTo1D $bias),
//       $qparams)),
//   $act_fn),

      $_state.addAttribute("value", value);
      $_state.addAttribute("s_param", s_param);
      $_state.addAttribute("compressed_data", compressed_data);
    }]>
  ];
}

def TFL_QuantizeOp: TFL_Op<"quantize", [
    FirstAttrDerivedResultType,
    SameOperandsAndResultShape, NoMemoryEffect]> {
  let summary = "Quantize operator";

  let description = [{

  );

  let results = (outs
    TF_FloatTensor:$output
  );

  TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
}

def TF_QuantizeV2Op : TF_Op<"QuantizeV2", [Pure]> {
  let summary = [{
Quantize the 'input' tensor of type float to 'output' tensor of type 'T'.
  }];

  let description = [{