diff options
| author |  Jianmin Chen <jmchen@google.com> | 2016-06-14 12:25:59 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-06-14 13:33:23 -0700 |
| commit | 0e96cccf3a0597e5b9d5a0971ec17dbe71659505 (patch) | |
| tree | 9c6da840599d4d95e44d500d6c38444b5dd3664b /tensorflow/python/tensorflow.i | |
| parent | 7adbf08ab85d1920cc56c3c43007b3a1750ce247 (diff) | |
Enable quantized training rewriting from the python side.
A simple example:
new_graph = pywrap_tensorflow.DoQuantizeTrainingOnGraphDef(8, sess.graph_def)
Then continue training with the new_graph.
Change: 124874712
Diffstat (limited to 'tensorflow/python/tensorflow.i')
| -rw-r--r-- | tensorflow/python/tensorflow.i | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/tensorflow.i b/tensorflow/python/tensorflow.i index 87397ead15..1bf63ff594 100644 --- a/tensorflow/python/tensorflow.i +++ b/tensorflow/python/tensorflow.i @@ -28,6 +28,7 @@ limitations under the License. %include "tensorflow/python/client/tf_session.i" %include "tensorflow/python/client/device_lib.i" +%include "tensorflow/python/client/quantize_training.i" %include "tensorflow/python/training/saver_io.i" %include "tensorflow/python/training/server_lib.i" |