/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/opt/
/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/fakefp16_transform.h (584B)
#pragma once #include #include #include #include #include #include namespace caffe2 { namespace opt { // Mapping from fp32 ops to fakefp16 ops TORCH_API std::unordered_map getFakeFp16OpMapping( bool use_fp16_acc = false, bool use_nnpi = false); TORCH_API void fakeFp16FuseOps(NetDef* net); // Transform normal fp32 operators to fakefp16 operators. TORCH_API void fakeFp16Transform(NetDef* net); } // namespace opt } // namespace caffe2