#define GEN_PASS_DEF_DEDUPBOUNDINPUTBINDINGPASS
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_savedmodel_passes.h.inc"

class DedupBoundInputBindingPass
    : public impl::DedupBoundInputBindingPassBase<DedupBoundInputBindingPass> {
  void runOnOperation() final;
};

void DedupBoundInputBindingPass::runOnOperation() {
  func::FuncOp func = getOperation();
  if (!mlir::tf_saved_model::IsExported(func)) return;

def FreezeAssetsPass : Pass<"tf-saved-model-freeze-assets", "ModuleOp"> {
  let summary = "Freeze tf_saved_model.asset's in func bodies.";
  let constructor = "mlir::tf_saved_model::CreateFreezeAssetsPass()";
}

def DedupBoundInputBindingPass : Pass<"tf-saved-model-dedup-bound-input-binding-pass", "mlir::func::FuncOp"> {
  let summary = "Remove duplicate 'tf_saved_model.bound_input' bindings.";