在 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 才能获得更改，但希望这能给您一个开始。