TensorFlow模型保存的一个坑

最近工作需要，使用tensorflow(r1.3)中contrib的某个库，根据tf官网的说法，go版本是只能调用py版本训练好的模型，而个人对go比较熟且线上环境用go作服务，所以想用go调python训练好的模型。

使用python import tf和tf.contrib相关库，使用tf.saved_model.builder.SavedModelBuilder保存模型，但使用go使用LoadSavedModel的时候，会报形如Not found Op type not registered 'CountExtremelyRandomStats' 的错误，具体的话是说在libtensorflow里面找不到CountExtremelyRandomStats 这个op。

尝试了很多方案，略过。最终发现问题是TensorFlow在BUILD target:libtensorflow.so的时候，根本没有把contrib相关内容打进去。修改根BUILD文件的libtensorflow.so部分如下：

cc_binary(
    name = "libtensorflow.so",
    linkopts = select({
        "//tensorflow:darwin": [
            "-Wl,-exported_symbols_list",  # This line must be directly followed by the exported_symbols.lds file
            "//tensorflow/c:exported_symbols.lds",
        ],
        "//tensorflow:windows": [],
        "//tensorflow:windows_msvc": [],
        "//conditions:default": [
            "-z defs",
            "-s",
            "-Wl,--version-script",  #  This line must be directly followed by the version_script.lds file
            "//tensorflow/c:version_script.lds",
        ],
    }),
    linkshared = 1,
    deps = [
        "//tensorflow/contrib:contrib_kernels",
        "//tensorflow/contrib:contrib_ops_op_lib",
        "//tensorflow/c:c_api",
        "//tensorflow/c:exported_symbols.lds",
        "//tensorflow/c:version_script.lds",
        "//tensorflow/core:tensorflow",
    ],
)

contrib那两行依赖是新加进去的，然后重新用bazel编译一个libtensorflow就可以了。期间困惑的是为什么python用pip装的就可以用contrib呢，观察tools/pip_packages/BUILD文件发现pip的包，是包含了所有contrib的依赖的。感觉官方团队对非python语言的重视还不够，没有明显说明这个东西，有人提了issue也没有人回应（我最后回应了这个op相关的一个issue：https://github.com/tensorflow/tensorflow/issues/14358）。