From d548cb4e811fc8a04dd10370c576441fc56b03f2 Mon Sep 17 00:00:00 2001 From: Suharsh Sivakumar Date: Mon, 19 Mar 2018 17:56:02 -0700 Subject: [PATCH] Add docstring pointing to tf.contrib.quantize. PiperOrigin-RevId: 189672549 --- tensorflow/python/training/quantize_training.i | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tensorflow/python/training/quantize_training.i b/tensorflow/python/training/quantize_training.i index 17ffcd6..fb5e47e 100644 --- a/tensorflow/python/training/quantize_training.i +++ b/tensorflow/python/training/quantize_training.i @@ -56,6 +56,11 @@ PyObject* DoQuantizeTrainingOnGraphDefHelper( %insert("python") %{ def do_quantize_training_on_graphdef(input_graph, num_bits): + """A 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. + """ from tensorflow.core.framework.graph_pb2 import GraphDef from tensorflow.python.framework import errors with errors.raise_exception_on_not_ok_status() as status: -- 2.7.4