1 # Copyright 2016 Google Inc. 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 """TF-GAN estimator module. 16 17 GANEstimator provides all the infrastructure support of a TensorFlow Estimator 18 with the feature support of TF-GAN. 19 """ 20 21 from __future__ import absolute_import 22 from __future__ import division 23 from __future__ import print_function 24 25 # Collapse `estimator` into a single namespace. 26 # pylint: disable=unused-import,wildcard-import 27 from tensorflow.contrib.gan.python.estimator.python import gan_estimator 28 from tensorflow.contrib.gan.python.estimator.python import head 29 from tensorflow.contrib.gan.python.estimator.python import latent_gan_estimator 30 from tensorflow.contrib.gan.python.estimator.python import stargan_estimator 31 from tensorflow.contrib.gan.python.estimator.python import tpu_gan_estimator 32 33 from tensorflow.contrib.gan.python.estimator.python.gan_estimator import * 34 from tensorflow.contrib.gan.python.estimator.python.head import * 35 from tensorflow.contrib.gan.python.estimator.python.latent_gan_estimator import * 36 from tensorflow.contrib.gan.python.estimator.python.stargan_estimator import * 37 from tensorflow.contrib.gan.python.estimator.python.tpu_gan_estimator import * 38 # pylint: enable=unused-import,wildcard-import 39 40 from tensorflow.python.util.all_util import remove_undocumented 41 42 _allowed_symbols = ([ 43 'gan_estimator', 44 'stargan_estimator', 45 'tpu_gan_estimator', 46 'latent_gan_estimator', 47 'head', 48 ] + gan_estimator.__all__ + stargan_estimator.__all__ + head.__all__ + 49 tpu_gan_estimator.__all__ + latent_gan_estimator.__all__) 50 remove_undocumented(__name__, _allowed_symbols) 51