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

include "mlir/IR/OpBase.td"
include "mlir/IR/PatternBase.td"
include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"

// TODO(jpienaar): Move this somewhere general.
class GetI64ScalarElementsAttr<int value> :

See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/

include "mlir/Pass/PassBase.td"

def ComposeUniformQuantizedTypePass : Pass<"compose-uniform-quantized-type", "ModuleOp"> {
  let summary = "Compose uniform quantized types in StableHLO.";

// TFLite legalization patterns

include "mlir/IR/OpBase.td"
include "mlir/IR/PatternBase.td"
include "mlir/Dialect/Arith/IR/ArithOps.td"
include "mlir/Dialect/Func/IR/FuncOps.td"
include "tensorflow/compiler/mlir/lite/ir/tfl_ops.td"
include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td"
include "tensorflow/compiler/mlir/lite/utils/utils.td"

def CreateEmptyBoolAttr : NativeCodeCall<"::mlir::BoolAttr()">;

include "mlir/Dialect/Shape/IR/ShapeBase.td"
include "mlir/IR/OpBase.td"
include "mlir/Interfaces/FunctionInterfaces.td"
include "mlir/Dialect/Quant/QuantOpsBase.td"
include "mlir/IR/SymbolInterfaces.td"
include "mlir/Interfaces/CallInterfaces.td"
include "mlir/Interfaces/ControlFlowInterfaces.td"
include "mlir/Interfaces/InferTypeOpInterface.td"
include "mlir/Interfaces/SideEffectInterfaces.td"

See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/

include "mlir/Pass/PassBase.td"

def OutlineCompositesPass: Pass<"outline-composites", "func::FuncOp"> {
  let summary = "Outlines specific patterns into composites.";
  let description = [{

//
// Predicates for types in the Quantization dialect.
//
//===----------------------------------------------------------------------===//

#ifndef QUANT_FORK_BASE
#define QUANT_FORK_BASE

include "mlir/IR/OpBase.td"

def QuantizationFork_Dialect : Dialect {
  let name = "quantfork";
  let cppNamespace = "::mlir::quantfork";
}

//
// Predicates for types in the Quantization dialect.
//
//===----------------------------------------------------------------------===//

#ifndef QUANTIZATION_BASE
#define QUANTIZATION_BASE

include "mlir/IR/OpBase.td"

def Quant_Dialect : Dialect {
  let name = "quantization";
  let cppNamespace = "::mlir::quant::ir";
}

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

include "tensorflow/compiler/mlir/tensorflow/transforms/optimize.td"
include "tensorflow/compiler/mlir/lite/ir/tfl_ops.td"
include "tensorflow/compiler/mlir/lite/utils/utils.td"

def FalseBoolAttr : AttrConstraint<CPred<"!$_self.getValue()">>;

def DenseElementsAttr : ElementsAttrBase<
  CPred<"$_self.isa<DenseElementsAttr>()">,

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

include "mlir/IR/PatternBase.td"
include "tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.td"
include "mhlo/IR/hlo_ops.td"


// See the function doc in the header file.
def GetPadOpType : NativeCodeCall<
  "GetPadOpType((*$0.begin()).getDefiningOp<mhlo::CompositeOp>())">;

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

#ifdef TFRT_FALLBACK_OPS
#else
#define TFRT_FALLBACK_OPS

include "mlir/Interfaces/SideEffectInterfaces.td"

def Fallback_Dialect : Dialect {
  let name = "tfrt_fallback";

  let description = [{
    The TFRT Fallback Dialect.
  }];

  let cppNamespace = "::tfrt::fallback";
}