diff options
author | Suharsh Sivakumar <suharshs@google.com> | 2017-03-29 18:17:36 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-29 19:31:58 -0700 |
commit | 20c86952373f61417a4a0829c57ff9640d91c177 (patch) | |
tree | bebd119ef41ad22f7bc38b7cf3d089b1c843d0f0 /tensorflow/python/tensorflow.i | |
parent | 3e239dc36147ef2d730ff4de50de59d9acfe0181 (diff) |
Make quantize training rewriter add variables to save and restore ops if a save op exists. This allows checkpointing of variables created by the rewriter.
Change: 151656461
Diffstat (limited to 'tensorflow/python/tensorflow.i')
-rw-r--r-- | tensorflow/python/tensorflow.i | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/tensorflow.i b/tensorflow/python/tensorflow.i index 09b4b20bcd..a0009031ac 100644 --- a/tensorflow/python/tensorflow.i +++ b/tensorflow/python/tensorflow.i @@ -29,9 +29,9 @@ 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/lib/io/file_io.i" +%include "tensorflow/python/training/quantize_training.i" %include "tensorflow/python/training/server_lib.i" %include "tensorflow/python/framework/python_op_gen.i" |