tf.compat.v1.train.do_quantize_training_on_graphdef

A general quantization scheme is being developed in tf.contrib.quantize. (deprecated)

Consider using that instead, though since it is in the tf.contrib namespace, it is not subject to backward compatibility guarantees.

Args
input_graph A GraphDef.
num_bits The number of bits for quantize training.
Returns
The graph with quantize training done.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/compat/v1/train/do_quantize_training_on_graphdef