llvm::SMLoc());
    SM.AddNewSourceBuffer(llvm::MemoryBuffer::getMemBuffer(check),
                          llvm::SMLoc());
    fc.readCheckFile(SM, llvm::StringRef(check));
    return fc.checkInput(SM, llvm::StringRef(input));
  });

    : public impl::InitTextFileToImportTestPassBase<
          InitTextFileToImportTestPass> {
 public:
  explicit InitTextFileToImportTestPass() {}

  StringRef getArgument() const final {
    return "tf-init-text-file-to-import-test";
  }

  StringRef getDescription() const final {
    return "generate a temporary file and invoke InitTextFileToImportPass";
  }

 private:
  void runOnOperation() override;
};

  }];

  let methods = [
    InterfaceMethod<
      [{Returns current operation data format (data layout).}],
      "StringRef", "getDataFormat", (ins)
    >,
    InterfaceMethod<
      [{Deprecated method that returns current operation data format (data layout).}],
      "StringRef", "data_format", (ins),
      /*methodBody=*/[{
        return $_op.getDataFormat();
    }]>,
    InterfaceMethod<

    return success();
  }

 private:
  StringRef saved_model_dir_;
};

void InitTextFileToImportPass::runOnOperation() {
  RewritePatternSet patterns(&getContext());
  MLIRContext* context = &getContext();
  func::FuncOp func = getOperation();

  patterns.add<ConvertInitializeTableFromTextFileV2>(
      context, StringRef(saved_model_dir_));

  LogicalResult RegisterKernel(llvm::StringRef kernel,
                               const KernelSpecs::Signature& signature,
                               const ScaleFn& fn, const ScaleDecomposeFn& dfn);

  // Adds the kernel spec with the scale constraint type for the kernel.
  LogicalResult RegisterKernel(llvm::StringRef kernel,
                               const KernelSpecs::Signature& signature,

#ifndef TENSORFLOW_COMPILER_MLIR_TF2XLA_TRANSFORMS_PASSES_H_
#define TENSORFLOW_COMPILER_MLIR_TF2XLA_TRANSFORMS_PASSES_H_

#include <memory>
#include <optional>

#include "llvm/ADT/StringRef.h"
#include "mlir/IR/MLIRContext.h"  // from @llvm-project
#include "mlir/IR/PatternMatch.h"  // from @llvm-project
#include "mlir/Pass/Pass.h"  // from @llvm-project
#include "mlir/Pass/PassRegistry.h"  // from @llvm-project

limitations under the License.
==============================================================================*/

#include "tensorflow/compiler/mlir/init_mlir.h"

#include "llvm/ADT/StringRef.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/PrettyStackTrace.h"
#include "tensorflow/core/platform/init_main.h"

static llvm::cl::extrahelp FlagSplittingHelp(R"(

  }
};

class SmuggleDisallowedOpsPass
    : public PassWrapper<SmuggleDisallowedOpsPass,
                         OperationPass<func::FuncOp>> {
 public:
  StringRef getArgument() const final { return "smuggle-disallowed-ops-pass"; }
  StringRef getDescription() const final {
    return "Smuggle disallowed ops via stablehlo.custom_calls";
  }
  void getDependentDialects(DialectRegistry& registry) const final {

namespace tfrt_compiler {

class TestCostAnalysis
    : public mlir::PassWrapper<TestCostAnalysis,
                               mlir::OperationPass<mlir::func::FuncOp>> {
  llvm::StringRef getArgument() const final {
    return "tfrt-test-cost-analysis";
  }
  llvm::StringRef getDescription() const final {
    return "Add remarks based on cost analysis for testing purpose.";
  }
  void runOnOperation() override {

#define TENSORFLOW_COMPILER_MLIR_TENSORFLOW_TRANSFORMS_TF_SAVED_MODEL_ASSET_SINKING_PASS_H_

#include <memory>

#include "absl/status/status.h"
#include "absl/strings/string_view.h"
#include "llvm/ADT/StringRef.h"
#include "mlir/IR/BuiltinOps.h"  // from @llvm-project
#include "mlir/Pass/Pass.h"  // from @llvm-project

namespace mlir {
namespace tf_saved_model {