46 lines
1.3 KiB
C++
46 lines
1.3 KiB
C++
//===--------- FrontendDialectTransformer.hpp - MLIR Operations -----------===//
|
|
//
|
|
// Copyright 2019 The IBM Research Authors.
|
|
//
|
|
// =============================================================================
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#pragma once
|
|
|
|
#include <fstream>
|
|
#include <functional>
|
|
#include <map>
|
|
#include <memory>
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "onnx/onnx_pb.h"
|
|
|
|
#include "src/Builder/FrontendDialectHelper.hpp"
|
|
|
|
namespace mlir {
|
|
class MLIRContext;
|
|
class OwningModuleRef;
|
|
} // namespace mlir
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Import a model into the ONNX MLIR dialect.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
namespace onnx_mlir {
|
|
/*!
|
|
* Import an ONNX model file into the ONNX Dialect.
|
|
* @param model_fname file name pointing to the onnx model protobuf.
|
|
* @return MLIR::module generated for the ONNX model.
|
|
*/
|
|
void ImportFrontendModelFile(std::string model_fname,
|
|
mlir::MLIRContext &context, mlir::OwningModuleRef &module);
|
|
|
|
/*!
|
|
* TODO: Import models into other extension dialects that cover the
|
|
* operations specific to other frameworks such as Tensorflow or Pytorch.
|
|
*/
|
|
} // namespace onnx_mlir
|