let summary = "Transform Einsum to other TF Ops for the supported variants";
  let constructor = "TF::CreateTransformEinsumPass()";
}

def LegalizeTFGToTFPass : Pass<"tfe-legalize-tfg", "ModuleOp"> {
  let summary = "Legalize from TFG to the TFE dialect";
  let constructor = "TF::CreateLegalizeTFGToTFEPass()";
}

def ReplicateToIslandPass : Pass<"tf-replicate-to-island", "mlir::func::FuncOp"> {

// RUN: tf-opt %s -tfl-legalize-tf --cse -split-input-file| FileCheck %s --dump-input=fail

func.func @add(%arg0: tensor<1xf32>, %arg1: tensor<1xf32>) -> tensor<1xf32> {
  %0 = "tf.Add"(%arg0, %arg1) : (tensor<1xf32>, tensor<1xf32>) -> tensor<1xf32>
  func.return %0: tensor<1xf32>

// CHECK-LABEL: add
// CHECK:  tfl.add %arg0, %arg1 {fused_activation_function = "NONE"} : tensor<1xf32>
// CHECK:  return
}

// -----

// ============================================================================
// The following functions tests example quantization patterns outputted from
// StableHLO Quantizer. These patterns should be legalized early directly
// to fused tflite ops.
// ============================================================================

// Tests that a simple per-channel quantized `stablehlo.dot_general` is properly

#include "tensorflow/core/lib/math/math_util.h"
#include "tensorflow/core/platform/status.h"
#include "tensorflow/core/util/padding.h"

namespace mlir {
namespace odml {
namespace {

#define DEBUG_TYPE "tf-legalize-hlo"

#define GEN_PASS_DEF_LEGALIZEHLOTOTFPASS
#include "tensorflow/compiler/mlir/lite/stablehlo/transforms/passes.h.inc"

class LegalizeHloToTf : public impl::LegalizeHloToTfPassBase<LegalizeHloToTf> {

                               : OpBuilder::atBlockEnd(body, listener);
    }
  }];

  // Canonicalizer wasn't defined for this one. In practise, we legalize the
  // tf.IfOp to scf.If op first and then legalize it to tfl.if to reduce
  // code redundancy.
}

def TFL_WhileOp : Op<TFL_Dialect, "while", [
    DeclareOpInterfaceMethods<LoopLikeOpInterface, ["isDefinedOutsideOfLoop"]>,

// Submitted by Rémi Mach <******@****.***>
loginline.app
loginline.dev
loginline.io
loginline.services
loginline.site

// Lokalized : https://lokalized.nl
// Submitted by Noah Taheij <noah@lokalized.nl>
servers.run

// Lõhmus Family, The
// Submitted by Heiki Lõhmus <hostmaster at lohmus dot me>
lohmus.me

// LubMAN UMCS Sp. z o.o : https://lubman.pl/

      Value value, const std::string& name, unsigned buffer_idx,
      const std::optional<BufferOffset<tflite::QuantizationParameters>>&
          quant_parameters);

  // TODO(b/137395003): Legalize tf.IfOp to TFLite dialect, and change the
  // following method to handle TFL::IfOp.
  BufferOffset<tflite::Operator> BuildIfOperator(
      mlir::TF::IfOp op, const std::vector<int32_t>& operands,

// 3) If `is_stateless` is false and no ordering token is present, then we treat
//    the op conservatively which means that different op instances need
//    dependencies. This is realized by always returning the same string ("")
//    in this case. In fact, we could return any string here, as long as it is
//    the same string for all op instances without ordering tokens.