mirror of https://github.com/llvm/torch-mlir
31 lines
928 B
C
31 lines
928 B
C
|
//===- ivalue_importer.h ----------------------------------------*- C++ -*-===//
|
||
|
|
//
|
||
|
|
// This file is licensed under a pytorch-style license
|
||
|
|
// See frontends/pytorch/LICENSE for license information.
|
||
|
|
//
|
||
|
|
//===----------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
#ifndef NPCOMP_FRONTENDS_PYTORCH_CSRC_IVALUE_IMPORTER_H
|
||
|
|
#define NPCOMP_FRONTENDS_PYTORCH_CSRC_IVALUE_IMPORTER_H
|
||
|
|
|
||
|
|
#include <memory>
|
||
|
|
|
||
|
|
#include "../pybind.h"
|
||
|
|
#include "func_builder.h"
|
||
|
|
|
||
|
|
#include "mlir-c/IR.h"
|
||
|
|
|
||
|
|
#include <torch/csrc/jit/api/compilation_unit.h>
|
||
|
|
#include <torch/csrc/jit/api/module.h>
|
||
|
|
#include <torch/csrc/jit/ir/ir.h>
|
||
|
|
|
||
|
|
namespace torch_mlir {
|
||
|
|
|
||
|
|
/// Main entry-point for importing torch IValue's .
|
||
|
|
/// Recursively imports `ivalue`, inserting operations at the end of `block`.
|
||
|
|
void importIValue(c10::IValue ivalue, MlirBlock block, MlirContext context);
|
||
|
|
|
||
|
|
} // namespace torch_mlir
|
||
|
|
|
||
|
|
#endif // NPCOMP_FRONTENDS_PYTORCH_CSRC_IVALUE_IMPORTER_H
|