o ?e3@sXdZddlmZddlmZddlmZddlmZe ddedgd d d Z dS) z!Quantize training for TensorFlow.) graph_pb2)"DoQuantizeTrainingOnGraphDefHelper) deprecation) tf_exportNzDGraphDef quantized training rewriter is deprecated in the long term.z&train.do_quantize_training_on_graphdef)v1cCs$t}t||}|||S)anA general quantization scheme is being developed in `tf.contrib.quantize`. 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. )rZGraphDefrZSerializeToStringZParseFromString)Z input_graphZnum_bitsgraphZresult_graph_stringrm/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/training/quantize_training.py do_quantize_training_on_graphdefs  r ) __doc__Ztensorflow.core.frameworkrZ+tensorflow.python._pywrap_quantize_trainingrZtensorflow.python.utilrZ tensorflow.python.util.tf_exportr deprecatedr rrrr s