module->setAttr("tf_saved_model.semantics",
                  UnitAttr::get(builder.getContext()));
  return absl::OkStatus();
}

std::unique_ptr<OperationPass<ModuleOp>> CreateAssetSinkingPass(
    llvm::StringRef saved_model_dir) {
  return std::make_unique<AssetSinkingPass>(saved_model_dir);
}

}  // namespace tf_saved_model