/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/opt
NameSizeModeActions
annotations.h21410644editdlrm
backend_cutting.h5860644editdlrm
backend_transformer_base.h27730644editdlrm
bound_shape_inferencer.h58840644editdlrm
converter.h30130644editdlrm
device.h4420644editdlrm
distributed.h11260644editdlrm
fakefp16_transform.h5840644editdlrm
fusion.h41410644editdlrm
glow_net_transform.h16250644editdlrm
mobile.h3880644editdlrm
onnxifi_op.h194970644editdlrm
onnxifi_transformer.h68590644editdlrm
onnx_convert.h12790644editdlrm
optimizer.h4020644editdlrm
optimize_ideep.h3510644editdlrm
passes.h26420644editdlrm
shape_info.h47840644editdlrm
tvm_transformer.h29920644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/caffe2/opt/mobile.h (388B)
#ifndef CAFFE2_OPT_MOBILE_H_ #define CAFFE2_OPT_MOBILE_H_ #include "caffe2/core/common.h" #include "nomnigraph/Representations/NeuralNet.h" namespace caffe2 { namespace opt { TORCH_API void addNNPACK(nom::repr::NNModule* nn, bool low_memory = false); TORCH_API void fuseNNPACKConvRelu(nom::repr::NNModule* nn); } // namespace opt } // namespace caffe2 #endif // CAFFE2_OPT_MOBILE_H_