FM / opt

/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/opt/

/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/opt

Name	Size	Mode	Actions
annotations.h	2141	0644	edit dl rm
backend_cutting.h	586	0644	edit dl rm
backend_transformer_base.h	2773	0644	edit dl rm
bound_shape_inferencer.h	5884	0644	edit dl rm
converter.h	3013	0644	edit dl rm
device.h	442	0644	edit dl rm
distributed.h	1126	0644	edit dl rm
fakefp16_transform.h	584	0644	edit dl rm
fusion.h	4141	0644	edit dl rm
glow_net_transform.h	1625	0644	edit dl rm
mobile.h	388	0644	edit dl rm
onnxifi_op.h	19497	0644	edit dl rm
onnxifi_transformer.h	6859	0644	edit dl rm
onnx_convert.h	1279	0644	edit dl rm
optimizer.h	402	0644	edit dl rm
optimize_ideep.h	351	0644	edit dl rm
passes.h	2642	0644	edit dl rm
shape_info.h	4784	0644	edit dl rm
tvm_transformer.h	2992	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/caffe2/opt/backend_cutting.h (586B)

#pragma once #include "caffe2/core/common.h" #include "caffe2/proto/caffe2_pb.h" #include "nomnigraph/Representations/NeuralNet.h" #include namespace caffe2 { namespace opt { struct CutResult { caffe2::NetDef net; int numberOfSubnets{0}; }; TORCH_API void DumpGraph(nom::repr::NNGraph* g, const std::string& fname); TORCH_API CutResult OptimizeForBackend( caffe2::NetDef& net, std::function supports, std::function transform_func, bool debug = false); } } // namespace caffe2