| #pragma once | |
| #include "torch/csrc/jit/ir.h" | |
| #include "torch/csrc/onnx/onnx.h" | |
| namespace torch { namespace jit { | |
| TORCH_API std::shared_ptr<Graph> ToONNX(std::shared_ptr<Graph>& state, ::torch::onnx::OperatorExportTypes operator_export_type); | |
| TORCH_API void BlockToONNX(Block* old_block, Block* new_block, ::torch::onnx::OperatorExportTypes operator_export_type, std::unordered_map<Value*, Value*> env); | |
| }} |