1955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# Copyright 2017 The TensorFlow Authors. All Rights Reserved. 2955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# 3955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# Licensed under the Apache License, Version 2.0 (the "License"); 4955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# you may not use this file except in compliance with the License. 5955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# You may obtain a copy of the License at 6955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# 7955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# http://www.apache.org/licenses/LICENSE-2.0 8955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# 9955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# Unless required by applicable law or agreed to in writing, software 10955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# distributed under the License is distributed on an "AS IS" BASIS, 11955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# See the License for the specific language governing permissions and 13955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# limitations under the License. 14955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# ============================================================================== 15955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar"""Functions for rewriting graphs for quantized training.""" 16955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar 17955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumarfrom __future__ import absolute_import 18955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumarfrom __future__ import division 19955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumarfrom __future__ import print_function 20955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar 21955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# pylint: disable=unused-import,wildcard-import,line-too-long 22955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumarfrom tensorflow.contrib.quantize.python.quantize_graph import * 23955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar# pylint: enable=unused-import,wildcard-import,line-too-long 24955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar 25955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumarfrom tensorflow.python.util.all_util import remove_undocumented 26955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar 27955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar_allowed_symbols = [ 28df2768c93b60fd60e353cebddc27de8390bebd4bSuharsh Sivakumar "create_eval_graph", 29955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar "create_training_graph", 3098c4a4efaf836e120505b4f1d52d7508802e004eSuharsh Sivakumar "experimental_create_eval_graph", 3198c4a4efaf836e120505b4f1d52d7508802e004eSuharsh Sivakumar "experimental_create_training_graph", 32955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar] 33955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumar 34955c525d416c163c9dd857e637b0476b112b0ea0Suharsh Sivakumarremove_undocumented(__name__, _allowed_symbols) 35