#include <optional>

#include "absl/strings/string_view.h"
#include "mlir/Pass/PassManager.h"  // from @llvm-project
#include "tensorflow/compiler/mlir/quantization/tensorflow/quantization_options.pb.h"

namespace tensorflow {
namespace quantization {

// mlir_dump_file_prefix is an optional field that is used for debugging to save
// mlir dump files.
void AddQuantizeQatPasses(mlir::OpPassManager &pm,

  // CHECK-NEXT:      buffer: 1,
  // CHECK-NEXT:      name: "arg0",
  // CHECK-NEXT:      quantization: {
  // CHECK-EMPTY:
  // CHECK-NEXT:      },
  // CHECK-NEXT:      has_rank: true
  // CHECK-NEXT:    }, {
  // CHECK-NEXT:      shape: [ 32 ],
  // CHECK-NEXT:      buffer: 2,
  // CHECK-NEXT:      name: "Const",
  // CHECK-NEXT:      quantization: {
  // CHECK-EMPTY:
  // CHECK-NEXT:      },
  // CHECK-NEXT:      has_rank: true

    }
  ],
  "subgraphs": [
    {
      "tensors": [
        {
          "shape": [1, 4],
          "name": "input",
          "quantization": {
          }
        },
        {
          "shape": [2, 2],
          "name": "output",
          "quantization": {
          }
        }
      ],
      "inputs": [0],
      "outputs": [1],
      "operators": [
        {
          "inputs": [ 0 ],

#include "mlir/Transforms/GreedyPatternRewriteDriver.h"  // from @llvm-project
#include "tensorflow/compiler/mlir/lite/quantization/ir/Passes.h"
#include "tensorflow/compiler/mlir/lite/quantization/ir/QuantOps.h"
#include "tensorflow/compiler/mlir/quantization/common/ir/FakeQuantSupport.h"
#include "tensorflow/compiler/mlir/quantization/common/ir/UniformSupport.h"

namespace mlir::quantfork {

#define GEN_PASS_DEF_QUANTCONVERTSIMULATEDQUANT

# ==============================================================================
from typing import Any

from tensorflow.compiler.mlir.quantization.tensorflow.calibrator import calibration_statistics_pb2
from tensorflow.compiler.mlir.quantization.tensorflow.python import py_function_lib
from tensorflow.compiler.mlir.quantization.tensorflow.python import representative_dataset as rd

# LINT.IfChange(quantize_qat_model)
def quantize_qat_model(

                        "dump_0001_tensorflow::quantization::mlir_dump_test"
                        "::ParentPass_before.mlir")));
  TF_EXPECT_OK(tsl::Env::Default()->FileExists(
      tsl::io::JoinPath(test_dir_,
                        "dump2_0001_tensorflow::quantization::mlir_dump_test"
                        "::NoOpPass_before.mlir")));
}
}  // namespace
}  // namespace quantization

// CHECK-NEXT:          name: "arg0",
// CHECK-NEXT:          quantization: {
// CHECK-EMPTY:
// CHECK-NEXT:          },
// CHECK-NEXT:          has_rank: true
// CHECK-NEXT:        }, {
// CHECK-NEXT:          shape: [ 4, 5 ],
// CHECK-NEXT:          buffer: 2,
// CHECK-NEXT:          name: "cast1",
// CHECK-NEXT:          quantization: {
// CHECK-EMPTY:
// CHECK-NEXT:          },

limitations under the License.
==============================================================================*/
#include "tensorflow/compiler/mlir/quantization/tensorflow/passes/tf_quant_ops.h"

#include "tensorflow/compiler/mlir/tensorflow/ir/tf_dialect.h"

#define GET_OP_CLASSES
#include "tensorflow/compiler/mlir/quantization/tensorflow/passes/tf_quant_ops.cc.inc"

namespace mlir {
namespace quant {

// CHECK-NEXT:       quantization: {
// CHECK-EMPTY:
// CHECK-NEXT:       },
// CHECK-NEXT:       has_rank: true
// CHECK-NEXT:     }, {
// CHECK-NEXT:       shape: [ 3, 2 ],
// CHECK-NEXT:       type: INT32,
// CHECK-NEXT:       buffer: 2,
// CHECK-NEXT:       name: "Const",
// STRIP:            buffer: 2,
// STRIP-NEXT:       name: "0",
// CHECK-NEXT:       quantization: {
// CHECK-EMPTY:
// CHECK-NEXT:       },

// CHECK-NEXT:       quantization: {
// CHECK-EMPTY:
// CHECK-NEXT:       },
// CHECK-NEXT:       has_rank: true
// CHECK-NEXT:     }, {
// CHECK-NEXT:       shape: [ 3, 2 ],
// CHECK-NEXT:       type: INT32,
// CHECK-NEXT:       buffer: 2,
// CHECK-NEXT:       name: "Const",
// STRIP:            buffer: 2,
// STRIP-NEXT:       name: "0",
// CHECK-NEXT:       quantization: {
// CHECK-EMPTY:
// CHECK-NEXT:       },