49 lines
1.6 KiB
C++
49 lines
1.6 KiB
C++
//===------------------ MLONNXOps.cpp - ONNX ML Operations ----------------===//
|
|
//
|
|
// Copyright 2019-2020 The IBM Research Authors.
|
|
//
|
|
// =============================================================================
|
|
//
|
|
// This file provides definition of ONNX ML dialect operations.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/Traits.h"
|
|
#include "mlir/IR/Block.h"
|
|
#include "mlir/IR/Builders.h"
|
|
#include "mlir/IR/Function.h"
|
|
#include "mlir/IR/IntegerSet.h"
|
|
#include "mlir/IR/Matchers.h"
|
|
#include "mlir/IR/Module.h"
|
|
#include "mlir/IR/OpImplementation.h"
|
|
#include "mlir/IR/PatternMatch.h"
|
|
#include "llvm/ADT/SetVector.h"
|
|
#include "llvm/ADT/SmallBitVector.h"
|
|
|
|
#include "MLONNXOps.hpp"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::OpTrait::util;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// MLONNXOpsDialect
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
/// Dialect creation, the instance will be owned by the context. This is the
|
|
/// point of registration of custom types and operations for the dialect.
|
|
MLONNXOpsDialect::MLONNXOpsDialect(mlir::MLIRContext *ctx)
|
|
: mlir::Dialect(getDialectNamespace(), ctx) {
|
|
addOperations<
|
|
#define GET_OP_LIST
|
|
#include "src/Dialect/MLONNX/MLONNXOps.cpp.inc"
|
|
>();
|
|
}
|
|
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// TableGen'd op method definitions
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#define GET_OP_CLASSES
|
|
#include "src/Dialect/MLONNX/MLONNXOps.cpp.inc"
|