在TensorFlow中执行伪量化的代码在哪里？

Question

我想在TensorFlow中创建一个自定义量化器（不是标准8位）。

我已经遍历了tensorflow\tensorflow\contrib\quantize\python中的代码，可以看到如何添加节点，但是我想修改tf.fake_quantize_with_min_max_vars函数如何计算输出。

我似乎找不到真正将32位累加并下采样到8位的代码。谁能指出我该代码的位置？

Answer 1

执行值的实际量化的代码在C ++中，在此功能中： https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/fake_quant_ops_functor.h#L79

修改并不是特别容易，因为您需要重建TensorFlow来获取更改，但是希望可以为您提供一个开始。