1 # Copyright 2017 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 """TFGAN features module. 16 17 This module includes support for virtual batch normalization, buffer replay, 18 conditioning, etc. 19 """ 20 21 from __future__ import absolute_import 22 from __future__ import division 23 from __future__ import print_function 24 25 # Collapse features into a single namespace. 26 # pylint: disable=unused-import,wildcard-import 27 from tensorflow.contrib.gan.python.features.python import clip_weights 28 from tensorflow.contrib.gan.python.features.python import conditioning_utils 29 from tensorflow.contrib.gan.python.features.python import random_tensor_pool 30 from tensorflow.contrib.gan.python.features.python import virtual_batchnorm 31 32 from tensorflow.contrib.gan.python.features.python.clip_weights import * 33 from tensorflow.contrib.gan.python.features.python.conditioning_utils import * 34 from tensorflow.contrib.gan.python.features.python.random_tensor_pool import * 35 from tensorflow.contrib.gan.python.features.python.virtual_batchnorm import * 36 # pylint: enable=unused-import,wildcard-import 37 38 from tensorflow.python.util.all_util import remove_undocumented 39 40 _allowed_symbols = clip_weights.__all__ 41 _allowed_symbols += conditioning_utils.__all__ 42 _allowed_symbols += random_tensor_pool.__all__ 43 _allowed_symbols += virtual_batchnorm.__all__ 44 remove_undocumented(__name__, _allowed_symbols) 45