Home | History | Annotate | Download | only in opt 
      1     # Copyright 2016 The TensorFlow Authors. All Rights Reserved.
      2 #
      3 # Licensed under the Apache License, Version 2.0 (the "License");
      4 # you may not use this file except in compliance with the License.
      5 # You may obtain a copy of the License at
      6 #
      7 #     http://www.apache.org/licenses/LICENSE-2.0
      8 #
      9 # Unless required by applicable law or agreed to in writing, software
     10 # distributed under the License is distributed on an "AS IS" BASIS,
     11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     12 # See the License for the specific language governing permissions and
     13 # limitations under the License.
     14 # ==============================================================================
     15 """A module containing optimization routines."""
     16 
     17 from __future__ import absolute_import
     18 from __future__ import division
     19 from __future__ import print_function
     20 
     21 # pylint: disable=wildcard-import
     22 from tensorflow.contrib.opt.python.training.addsign import *
     23 from tensorflow.contrib.opt.python.training.drop_stale_gradient_optimizer import *
     24 from tensorflow.contrib.opt.python.training.external_optimizer import *
     25 from tensorflow.contrib.opt.python.training.lazy_adam_optimizer import *
     26 from tensorflow.contrib.opt.python.training.moving_average_optimizer import *
     27 from tensorflow.contrib.opt.python.training.multitask_optimizer_wrapper import *
     28 from tensorflow.contrib.opt.python.training.nadam_optimizer import *
     29 from tensorflow.contrib.opt.python.training.powersign import *
     30 from tensorflow.contrib.opt.python.training.variable_clipping_optimizer import *
     31 from tensorflow.contrib.opt.python.training.elastic_average_optimizer import *
     32 from tensorflow.contrib.opt.python.training.model_average_optimizer import *
     33 # pylint: enable=wildcard-import
     34 
     35 from tensorflow.python.util.all_util import remove_undocumented
     36 
     37 
     38 _allowed_symbols = [
     39     'PowerSignOptimizer',
     40     'AddSignOptimizer',
     41     'DelayCompensatedGradientDescentOptimizer',
     42     'DropStaleGradientOptimizer',
     43     'ExternalOptimizerInterface',
     44     'LazyAdamOptimizer',
     45     'NadamOptimizer',
     46     'MovingAverageOptimizer',
     47     'ScipyOptimizerInterface',
     48     'VariableClippingOptimizer',
     49     'MultitaskOptimizerWrapper',
     50     'clip_gradients_by_global_norm',
     51     'ElasticAverageOptimizer',
     52     'ElasticAverageCustomGetter',
     53     'ModelAverageOptimizer',
     54     'ModelAverageCustomGetter'
     55 ]
     56 
     57 remove_undocumented(__name__, _allowed_symbols)
     58