diff options
author | 2017-12-01 16:38:52 -0800 | |
---|---|---|
committer | 2017-12-01 16:46:20 -0800 | |
commit | e30b0babce133631b19de1fd7bacc84c884d6f55 (patch) | |
tree | 2b034ed6283511f192760ed8c84e7870c7d29f9f /tensorflow/python/training/quantize_training.i | |
parent | 4173ac695eb7b41fc1885b08ed52904dbf2278c3 (diff) |
Adding tf_export decorators to public API functions defined in swig files.
PiperOrigin-RevId: 177657932
Diffstat (limited to 'tensorflow/python/training/quantize_training.i')
-rw-r--r-- | tensorflow/python/training/quantize_training.i | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/python/training/quantize_training.i b/tensorflow/python/training/quantize_training.i index 40c6076973..17ffcd6e07 100644 --- a/tensorflow/python/training/quantize_training.i +++ b/tensorflow/python/training/quantize_training.i @@ -65,6 +65,9 @@ def do_quantize_training_on_graphdef(input_graph, num_bits): graph.ParseFromString(result_graph_string) return graph + +do_quantize_training_on_graphdef._tf_api_names = [ + 'train.do_quantize_training_on_graphdef'] %} %unignoreall |