.Attr("file_name: string")
    .Attr("T: type")
    .Attr("enabled: bool")
    .Attr("func_name: string")
    .Attr("node_name: string")
    .SetIsStateful();

class DumpTensorOp : public OpKernel {
 public:
  explicit DumpTensorOp(OpKernelConstruction* ctx) : OpKernel(ctx) {
    string log_dir_path;
    string file_name;
    string func_name;
    string node_name;

        tensorflow::io::JoinPath(log_dir_path_, f_attr.getValue());

    // Attach DumpTensorOp to its output layer.
    SmallVector<NamedAttribute> dump_attributes =
        CreateDumpAttributes(rewriter, folder_name, file_name,
                             /*enabled=*/true, func_name, node_name);
    rewriter.create<TF::DumpTensorOp>(op->getLoc(), TypeRange{}, result,
                                      dump_attributes);

namespace stablehlo::quantization {

void DisableDebugging(mlir::ModuleOp module_op) {
  module_op.walk(
      [](mlir::TF::DumpTensorOp dump_op) { dump_op.setEnabled(false); });
}

void ChangeToQuantizedFilename(mlir::ModuleOp module_op) {
  module_op.walk([](mlir::TF::DumpTensorOp dump_op) {
    dump_op.setFileName("quantized_tensor_data.pb");
  });
}

  repeated QuantizationSpec specs = 1;
}

// Configuration for quantization debugger.
// The debugging model assumes it should be run on CPU based server, since the
// model contains TF::DumpTensorOp.
// NEXT ID: 4
message DebuggerConfig {
  // Type of quantization debugger. Depending on the type, inputs and outputs
  // are wired differently.
  // NEXT ID: 4
  enum DebuggerType {

      meta_graph = saved_model_loader.load(sess, tags, export_dir=model_dir)
      signature_def = meta_graph.signature_def[signature_key]

      # DumpTensorOp only works in graph mode.
      # Execute the model using session to run DumpTensorOp.
      output_tensor_names = [
          output_tensor_info.name
          for output_tensor_info in signature_def.outputs.values()
      ]