More code fakefp16 mapping unification

Summary: ATT

Reviewed By: amylittleyang

Differential Revision: D19597036

fbshipit-source-id: deed61945884fb4b01d058f3c72c75f5a937a41c

Author: Yinghai Lu

caffe2/opt/custom/fakefp16_transform.h

@@ -12,7 +12,7 @@ namespace caffe2 {
 namespace opt {
 
 // Mapping from fp32 ops to fakefp16 ops
-std::unordered_map<std::string, std::string> getFakeFp16OpMapping(
+CAFFE2_API std::unordered_map<std::string, std::string> getFakeFp16OpMapping(
     bool use_fp16_acc = false,
     bool use_nnpi = false);
 

caffe2/python/pybind_state.cc

@@ -21,6 +21,7 @@
 #include "caffe2/onnx/helper.h"
 #include "caffe2/onnx/onnx_exporter.h"
 #include "caffe2/opt/converter.h"
+#include "caffe2/opt/custom/fakefp16_transform.h"
 #include "caffe2/opt/fusion.h"
 #include "caffe2/opt/mobile.h"
 #include "caffe2/opt/onnxifi_transformer.h"
@@ -1706,6 +1707,9 @@ void addGlobalMethods(py::module& m) {
     new_proto.SerializeToString(&out);
     return py::bytes(out);
   });
+  m.def("get_fakefp16_mapping", [](bool use_fp16_acc, bool use_nnpi) {
+    return caffe2::opt::getFakeFp16OpMapping(use_fp16_acc, use_nnpi);
+  });
   m.def(
       "onnxifi",
       [](const py::bytes& pred_net_str,