torch/csrc/onnx/export.h

#pragma once

#include "torch/csrc/jit/ir.h"

namespace torch { namespace jit {

// Exports a graph to ONNX
std::string ExportGraph(std::shared_ptr<Graph>& graph,
                        const std::unordered_map<void*, Node*>& buffer_map,
                        const std::vector<at::Tensor> & initializers,
                        bool verbose=false);

}}