caffe2_importer.cpp File Reference

#include "caffe2_importer.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2_op_types.h"
#include "caffe2_op_creator.h"
#include "caffe2_proto_helper.h"
#include "mir/ops/InputOp.h"
#include "mir/ops/OutputOp.h"
#include <google/protobuf/io/zero_copy_stream_impl.h>
#include <google/protobuf/io/coded_stream.h>
#include <fcntl.h>
#include <cassert>
#include <cerrno>
#include <cstring>
#include <memory>
#include <stdexcept>
#include <utility>
#include <set>

Go to the source code of this file.

Namespaces
namespace	mir_caffe2

Functions
std::unique_ptr< mir::Graph >	mir_caffe2::loadModel (std::string predict_net, std::string init_net, const std::vector< std::vector< int > > &input_shapes)