text: // Copyright 2019 Google LLC
  //
  // Licensed under the Apache License, Version 2.0 (the "License");
  // you may not use this file except in compliance with the License.
  // You may obtain a copy of the License at
  //
  //      https://www.apache.org/licenses/LICENSE-2.0
  //
  // Unless required by applicable law or agreed to in writing, software
  // distributed under the License is distributed on an "AS IS" BASIS,
  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  // See the License for the specific language governing permissions and
  // limitations under the License.
  
  #include "integrations/tensorflow/compiler/Passes.h"
  
  #include "integrations/tensorflow/compiler/dialect/tf_strings/conversion/convert_tf_to_tf_strings.h"
  #include "integrations/tensorflow/compiler/dialect/tf_tensorlist/conversion/convert_tf_to_tf_tensorlist.h"
  #include "iree/compiler/Dialect/Shape/Conversion/Passes.h"
  #include "iree/compiler/Dialect/Shape/Transforms/Passes.h"
  #include "mlir/Pass/PassManager.h"
  #include "mlir/Pass/PassRegistry.h"
  #include "mlir/Transforms/Passes.h"
  
  namespace mlir {
  namespace iree_compiler {
  
  // All IREE-specific passes that lower TF representations before reaching the
  // IREE core should go here.
  void createIreeTfImportPipeline(OpPassManager &pm) {
    ////////////////////////////////////////////////////////////////////////////
    // Lowering shape-related constructs.
    ////////////////////////////////////////////////////////////////////////////
    pm.addPass(Shape::createConvertHLOToShapePass());
    pm.addPass(Shape::createConvertShapeToShapexPass());
    // Clean up trivial redundancies.
    pm.addPass(createCanonicalizerPass());
  
    ////////////////////////////////////////////////////////////////////////////
    // Lowering TensorList-related parts of tf dialect to tf_tensorlist dialect.
    ////////////////////////////////////////////////////////////////////////////
    pm.addPass(tf_tensorlist::createConvertTfToTfTensorList());
  
    ////////////////////////////////////////////////////////////////////////////
    // Lowering strings-related parts of tf dialect to tf_strings dialect.
    ////////////////////////////////////////////////////////////////////////////
    pm.addPass(tf_strings::createConvertTfToTfStrings());
  
    ////////////////////////////////////////////////////
    // Lowering tf_saved_model dialect to IREE dialects.
    ////////////////////////////////////////////////////
    // First, eliminate tf_saved_model.global_tensor's and corresponding
    // tf_saved_model.bound_input's.
    pm.addPass(createTFSavedModelLowerGlobalTensors());
  
    // Lower exported functions.
    //
    // This pass must run second because:
    // - It assumes that tf_saved_model.bound_inputs have been eliminated
    // - It removes tf_saved_model.semantics from the module, which we can only
    //   do at the very end.
    pm.addPass(createTFSavedModelLowerExportedFunctions());
  }
  
  static mlir::PassPipelineRegistration<> pipeline(
      "iree-tf-import-pipeline",
      "Run IREE-specific passes for importing TF code into IREE.",
      createIreeTfImportPipeline);
  
  }  // namespace iree_compiler
  }  // namespace mlir
  