Name	Size	Mode	Actions
InferenceGraph.h	833	0644	editdlrm
predictor.h	1810	0644	editdlrm
predictor_config.h	1934	0644	editdlrm
predictor_utils.h	802	0644	editdlrm
ThreadLocalPtr.h	4111	0644	editdlrm
transforms.h	241	0644	editdlrm

#pragma once #include "caffe2/core/workspace.h" namespace caffe2 { /** * This struct stores information about the inference graph which defines * underlying math of BlackBoxPredictor. Other parts of it such as various * threading optimizations don't belong here. */ struct InferenceGraph { std::unique_ptr predict_init_net_def; // shared_ptr allows to share NetDef with its operators on each of the threads // without memory replication. Note that predict_init_net_def_ could be stored // by value as its operators are discarded immidiatly after use (via // RunNetOnce) std::shared_ptr predict_net_def; std::vector input_names; std::vector output_names; std::vector parameter_names; bool predictor_net_ssa_rewritten{false}; }; } // namespace caffe2