netdef_converter.h

#pragma once
#include <caffe2/proto/caffe2_pb.h>
#include <torch/csrc/jit/ir.h>
#include <unordered_map>

namespace torch {
namespace jit {

void convertNetDefToIR(
    const caffe2::NetDef& net,
    Graph* graph,
    std::unordered_map<std::string, Value*>* valueMapPtr = nullptr,
    const std::string& prefix = "");

void convertIRToNetDef(
    caffe2::NetDef* net,
    const Graph& graph,
    const std::string& prefix = "");

} // namespace jit
} // namespace torch