namespace mlir {
namespace TF {
namespace {

#define GEN_PASS_DEF_LOWERQUANTIZEDPASS
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc"

class LowerQuantizedPass
    : public impl::LowerQuantizedPassBase<LowerQuantizedPass> {
 public:
  void runOnOperation() override {
    RewritePatternSet patterns(&getContext());
    mlir::TF::PopulateLoweringQuantizedPatterns(&getContext(), &patterns);

  let dependentDialects = [
    "mhlo::MhloDialect",
    "tf_device::TensorFlowDeviceDialect"
  ];

  let constructor = "TFDevice::CreateReplicateInvariantOpHoistingPass()";
}

def LowerQuantizedPass : Pass<"tf-lower-quantized", "mlir::func::FuncOp"> {
  let summary = "Lowers ops that require quantized input or output.";

  let description = [{