/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/predictor
NameSizeModeActions
InferenceGraph.h8330644editdlrm
predictor.h18100644editdlrm
predictor_config.h19340644editdlrm
predictor_utils.h8020644editdlrm
ThreadLocalPtr.h41110644editdlrm
transforms.h2410644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/caffe2/predictor/transforms.h (241B)
#pragma once #include "caffe2/core/net.h" #include "caffe2/core/workspace.h" #include "caffe2/predictor/InferenceGraph.h" namespace caffe2 { void RemoveOpsByType(InferenceGraph& graph_, const std::string& op_type); } // namespace caffe2