| # Files for using TF-GAN framework. |
| |
| load("//tensorflow:tensorflow.bzl", "py_test") |
| |
| package( |
| default_visibility = [ |
| "//tensorflow:__subpackages__", |
| ], |
| licenses = ["notice"], # Apache 2.0 |
| ) |
| |
| exports_files(["LICENSE"]) |
| |
| py_library( |
| name = "gan", |
| srcs = [ |
| "__init__.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":estimator", |
| ":eval", |
| ":features", |
| ":losses", |
| ":namedtuples", |
| ":train", |
| "//tensorflow/python:util", |
| ], |
| ) |
| |
| py_library( |
| name = "namedtuples", |
| srcs = ["python/namedtuples.py"], |
| srcs_version = "PY2AND3", |
| ) |
| |
| py_library( |
| name = "train", |
| srcs = ["python/train.py"], |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":losses", |
| ":namedtuples", |
| "//tensorflow/contrib/framework:framework_py", |
| "//tensorflow/contrib/slim:learning", |
| "//tensorflow/contrib/training:training_py", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:check_ops", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:init_ops", |
| "//tensorflow/python:random_ops", |
| "//tensorflow/python:training", |
| "//tensorflow/python:training_util", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python/ops/losses", |
| ], |
| ) |
| |
| py_test( |
| name = "train_test", |
| srcs = ["python/train_test.py"], |
| python_version = "PY2", |
| shard_count = 50, |
| srcs_version = "PY2AND3", |
| tags = [ |
| "no_oss", |
| "notsan", |
| ], |
| deps = [ |
| ":namedtuples", |
| ":random_tensor_pool", |
| ":train", |
| "//tensorflow/contrib/framework:framework_py", |
| "//tensorflow/contrib/layers:layers_py", |
| "//tensorflow/contrib/slim:learning", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:constant_op", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:random_ops", |
| "//tensorflow/python:random_seed", |
| "//tensorflow/python:training", |
| "//tensorflow/python:training_util", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python:variables", |
| "//tensorflow/python/ops/distributions", |
| "//third_party/py/numpy", |
| "@absl_py//absl/testing:parameterized", |
| ], |
| ) |
| |
| py_library( |
| name = "eval", |
| srcs = ["python/eval/__init__.py"], |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":classifier_metrics", |
| ":eval_utils", |
| ":sliced_wasserstein", |
| ":summaries", |
| "//tensorflow/python:util", |
| ], |
| ) |
| |
| py_library( |
| name = "estimator", |
| srcs = ["python/estimator/__init__.py"], |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":gan_estimator", |
| ":head", |
| ":latent_gan_estimator", |
| ":stargan_estimator", |
| ":tpu_gan_estimator", |
| "//tensorflow/python:util", |
| ], |
| ) |
| |
| py_library( |
| name = "losses", |
| srcs = ["python/losses/__init__.py"], |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":losses_impl", |
| ":tuple_losses", |
| "//tensorflow/python:util", |
| ], |
| ) |
| |
| py_library( |
| name = "features", |
| srcs = ["python/features/__init__.py"], |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":clip_weights", |
| ":conditioning_utils", |
| ":random_tensor_pool", |
| ":spectral_normalization", |
| ":virtual_batchnorm", |
| "//tensorflow/python:util", |
| ], |
| ) |
| |
| py_library( |
| name = "losses_impl", |
| srcs = ["python/losses/python/losses_impl.py"], |
| srcs_version = "PY2AND3", |
| deps = [ |
| "//tensorflow/contrib/framework:framework_py", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:clip_ops", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:gradients_impl", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:random_ops", |
| "//tensorflow/python:summary", |
| "//tensorflow/python:tensor_util", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python/ops/losses", |
| ], |
| ) |
| |
| py_test( |
| name = "losses_impl_test", |
| srcs = ["python/losses/python/losses_impl_test.py"], |
| python_version = "PY2", |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":losses_impl", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:clip_ops", |
| "//tensorflow/python:constant_op", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:random_ops", |
| "//tensorflow/python:random_seed", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python:variables", |
| "//tensorflow/python/ops/distributions", |
| "//tensorflow/python/ops/losses", |
| ], |
| ) |
| |
| py_library( |
| name = "tuple_losses", |
| srcs = [ |
| "python/losses/python/losses_wargs.py", |
| "python/losses/python/tuple_losses.py", |
| "python/losses/python/tuple_losses_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":losses_impl", |
| ":namedtuples", |
| "//tensorflow/python:util", |
| ], |
| ) |
| |
| py_test( |
| name = "tuple_losses_test", |
| srcs = ["python/losses/python/tuple_losses_test.py"], |
| python_version = "PY2", |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":losses_impl", |
| ":namedtuples", |
| ":tuple_losses", |
| "//tensorflow/contrib/layers:layers_py", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:constant_op", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python:variables", |
| "//third_party/py/numpy", |
| ], |
| ) |
| |
| py_library( |
| name = "conditioning_utils", |
| srcs = [ |
| "python/features/python/conditioning_utils.py", |
| "python/features/python/conditioning_utils_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| "//tensorflow/contrib/layers:layers_py", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:embedding_ops", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:tensor_util", |
| "//tensorflow/python:util", |
| "//tensorflow/python:variable_scope", |
| ], |
| ) |
| |
| py_test( |
| name = "conditioning_utils_test", |
| srcs = ["python/features/python/conditioning_utils_test.py"], |
| python_version = "PY2", |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":conditioning_utils", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:dtypes", |
| ], |
| ) |
| |
| py_library( |
| name = "random_tensor_pool", |
| srcs = [ |
| "python/features/python/random_tensor_pool.py", |
| "python/features/python/random_tensor_pool_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:control_flow_ops", |
| "//tensorflow/python:data_flow_ops", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:random_ops", |
| "//tensorflow/python:util", |
| ], |
| ) |
| |
| py_test( |
| name = "random_tensor_pool_test", |
| srcs = ["python/features/python/random_tensor_pool_test.py"], |
| python_version = "PY2", |
| shard_count = 6, |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":random_tensor_pool", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:constant_op", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:framework_ops", |
| "//third_party/py/numpy", |
| ], |
| ) |
| |
| py_library( |
| name = "virtual_batchnorm", |
| srcs = [ |
| "python/features/python/virtual_batchnorm.py", |
| "python/features/python/virtual_batchnorm_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:init_ops", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:nn", |
| "//tensorflow/python:tensor_shape", |
| "//tensorflow/python:tensor_util", |
| "//tensorflow/python:util", |
| "//tensorflow/python:variable_scope", |
| ], |
| ) |
| |
| py_test( |
| name = "virtual_batchnorm_test", |
| srcs = ["python/features/python/virtual_batchnorm_test.py"], |
| python_version = "PY2", |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":virtual_batchnorm", |
| "//tensorflow/contrib/framework:framework_py", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:constant_op", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:layers", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:nn", |
| "//tensorflow/python:random_ops", |
| "//tensorflow/python:random_seed", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python:variables", |
| "//third_party/py/numpy", |
| ], |
| ) |
| |
| py_library( |
| name = "clip_weights", |
| srcs = [ |
| "python/features/python/clip_weights.py", |
| "python/features/python/clip_weights_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| "//tensorflow/contrib/opt:opt_py", |
| "//tensorflow/python:util", |
| ], |
| ) |
| |
| py_test( |
| name = "clip_weights_test", |
| srcs = ["python/features/python/clip_weights_test.py"], |
| python_version = "PY2", |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":clip_weights", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:training", |
| "//tensorflow/python:variables", |
| ], |
| ) |
| |
| py_library( |
| name = "classifier_metrics", |
| srcs = [ |
| "python/eval/python/classifier_metrics.py", |
| "python/eval/python/classifier_metrics_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| "//tensorflow/contrib/layers:layers_py", |
| "//tensorflow/core:protos_all_py", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:framework", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:functional_ops", |
| "//tensorflow/python:image_ops", |
| "//tensorflow/python:linalg_ops", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:nn", |
| "//tensorflow/python:nn_ops", |
| "//tensorflow/python:platform", |
| "//tensorflow/python:util", |
| "@six_archive//:six", |
| ], |
| ) |
| |
| py_test( |
| name = "classifier_metrics_test", |
| srcs = ["python/eval/python/classifier_metrics_test.py"], |
| python_version = "PY2", |
| srcs_version = "PY2AND3", |
| tags = [ |
| "no_pip", |
| "no_windows", |
| ], |
| deps = [ |
| ":classifier_metrics", |
| "//tensorflow/core:protos_all_py", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:variables", |
| "//third_party/py/numpy", |
| "@absl_py//absl/testing:parameterized", |
| ], |
| ) |
| |
| py_library( |
| name = "eval_utils", |
| srcs = [ |
| "python/eval/python/eval_utils.py", |
| "python/eval/python/eval_utils_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:util", |
| ], |
| ) |
| |
| py_test( |
| name = "eval_utils_test", |
| srcs = ["python/eval/python/eval_utils_test.py"], |
| python_version = "PY2", |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":eval_utils", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| ], |
| ) |
| |
| py_library( |
| name = "summaries", |
| srcs = [ |
| "python/eval/python/summaries.py", |
| "python/eval/python/summaries_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":eval_utils", |
| ":namedtuples", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:functional_ops", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:summary", |
| "//tensorflow/python:util", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python/ops/losses", |
| ], |
| ) |
| |
| py_test( |
| name = "summaries_test", |
| srcs = ["python/eval/python/summaries_test.py"], |
| python_version = "PY2", |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":namedtuples", |
| ":summaries", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:summary", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python:variables", |
| ], |
| ) |
| |
| py_library( |
| name = "head", |
| srcs = [ |
| "python/estimator/python/head.py", |
| "python/estimator/python/head_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":namedtuples", |
| ":train", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:util", |
| "//tensorflow/python/estimator:estimator_py", |
| ], |
| ) |
| |
| py_test( |
| name = "head_test", |
| srcs = ["python/estimator/python/head_test.py"], |
| python_version = "PY2", |
| shard_count = 1, |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":head", |
| ":namedtuples", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:training", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python/estimator:estimator_py", |
| ], |
| ) |
| |
| py_library( |
| name = "gan_estimator", |
| srcs = [ |
| "python/estimator/python/gan_estimator.py", |
| "python/estimator/python/gan_estimator_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":namedtuples", |
| ":summaries", |
| ":train", |
| "//tensorflow/contrib/framework:framework_py", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:metrics", |
| "//tensorflow/python:util", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python/estimator:estimator_py", |
| ], |
| ) |
| |
| py_test( |
| name = "gan_estimator_test", |
| srcs = ["python/estimator/python/gan_estimator_test.py"], |
| python_version = "PY2", |
| shard_count = 1, |
| srcs_version = "PY2AND3", |
| tags = [ |
| "no_oss", |
| "notsan", |
| ], |
| deps = [ |
| ":gan_estimator", |
| ":namedtuples", |
| ":tuple_losses", |
| "//tensorflow/contrib/layers:layers_py", |
| "//tensorflow/contrib/learn", |
| "//tensorflow/core:protos_all_py", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:errors", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:metrics", |
| "//tensorflow/python:parsing_ops", |
| "//tensorflow/python:summary", |
| "//tensorflow/python:tensor_shape", |
| "//tensorflow/python:training", |
| "//tensorflow/python:training_util", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python/estimator", |
| "//tensorflow/python/estimator:model_fn", |
| "//tensorflow/python/estimator:numpy_io", |
| "//third_party/py/numpy", |
| "@absl_py//absl/testing:parameterized", |
| "@six_archive//:six", |
| ], |
| ) |
| |
| py_library( |
| name = "stargan_estimator", |
| srcs = [ |
| "python/estimator/python/stargan_estimator.py", |
| "python/estimator/python/stargan_estimator_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":namedtuples", |
| ":summaries", |
| ":train", |
| "//tensorflow/contrib/framework:framework_py", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:metrics", |
| "//tensorflow/python:util", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python/estimator:estimator_py", |
| ], |
| ) |
| |
| py_test( |
| name = "stargan_estimator_test", |
| srcs = ["python/estimator/python/stargan_estimator_test.py"], |
| python_version = "PY2", |
| shard_count = 1, |
| srcs_version = "PY2AND3", |
| tags = ["notsan"], |
| deps = [ |
| ":namedtuples", |
| ":stargan_estimator", |
| "//tensorflow/contrib/layers:layers_py", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:metrics", |
| "//tensorflow/python:summary", |
| "//tensorflow/python:training", |
| "//tensorflow/python:training_util", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python/estimator:model_fn", |
| "//tensorflow/python/estimator:numpy_io", |
| "//third_party/py/numpy", |
| "@absl_py//absl/testing:parameterized", |
| "@six_archive//:six", |
| ], |
| ) |
| |
| py_library( |
| name = "tpu_gan_estimator", |
| srcs = [ |
| "python/estimator/python/tpu_gan_estimator.py", |
| "python/estimator/python/tpu_gan_estimator_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":gan_estimator", |
| ":namedtuples", |
| ":train", |
| "//tensorflow/contrib/tpu:tpu_estimator", |
| "//tensorflow/contrib/tpu:tpu_lib", |
| "//tensorflow/contrib/training:training_py", |
| "//tensorflow/python:control_flow_ops", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:metrics", |
| "//tensorflow/python:util", |
| "//tensorflow/python/estimator:model_fn", |
| "//tensorflow/python/ops/losses", |
| ], |
| ) |
| |
| py_test( |
| name = "tpu_gan_estimator_test", |
| srcs = ["python/estimator/python/tpu_gan_estimator_test.py"], |
| python_version = "PY2", |
| shard_count = 11, |
| srcs_version = "PY2AND3", |
| tags = ["notsan"], |
| deps = [ |
| ":namedtuples", |
| ":tpu_gan_estimator", |
| ":tuple_losses", |
| "//tensorflow/contrib/layers:layers_py", |
| "//tensorflow/contrib/tpu:tpu_estimator", |
| "//tensorflow/contrib/tpu:tpu_lib", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:errors", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:metrics", |
| "//tensorflow/python:summary", |
| "//tensorflow/python:tensor_shape", |
| "//tensorflow/python:training", |
| "//tensorflow/python:training_util", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python/data/ops:dataset_ops", |
| "//tensorflow/python/estimator", |
| "//tensorflow/python/estimator:model_fn", |
| "//third_party/py/numpy", |
| "@absl_py//absl/testing:parameterized", |
| "@six_archive//:six", |
| ], |
| ) |
| |
| py_library( |
| name = "latent_gan_estimator", |
| srcs = [ |
| "python/estimator/python/latent_gan_estimator.py", |
| "python/estimator/python/latent_gan_estimator_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":train", |
| "//tensorflow/python:clip_ops", |
| "//tensorflow/python:gradients", |
| "//tensorflow/python:random_ops", |
| "//tensorflow/python:summary", |
| "//tensorflow/python:training_util", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python/estimator:estimator_py", |
| ], |
| ) |
| |
| py_test( |
| name = "latent_gan_estimator_test", |
| srcs = [ |
| "python/estimator/python/latent_gan_estimator_test.py", |
| ], |
| python_version = "PY2", |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":latent_gan_estimator", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:training", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python/estimator:run_config", |
| "//tensorflow/python/ops/losses", |
| ], |
| ) |
| |
| py_library( |
| name = "sliced_wasserstein", |
| srcs = [ |
| "python/eval/python/sliced_wasserstein.py", |
| "python/eval/python/sliced_wasserstein_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:constant_op", |
| "//tensorflow/python:linalg_ops", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:nn", |
| "//tensorflow/python:nn_ops", |
| "//tensorflow/python:random_ops", |
| "//tensorflow/python:script_ops", |
| "//tensorflow/python:util", |
| "//third_party/py/numpy", |
| ], |
| ) |
| |
| py_test( |
| name = "sliced_wasserstein_test", |
| srcs = ["python/eval/python/sliced_wasserstein_test.py"], |
| python_version = "PY2", |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":sliced_wasserstein", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:random_ops", |
| "//third_party/py/numpy", |
| ], |
| ) |
| |
| py_library( |
| name = "spectral_normalization", |
| srcs = [ |
| "python/features/python/spectral_normalization.py", |
| "python/features/python/spectral_normalization_impl.py", |
| ], |
| srcs_version = "PY2AND3", |
| deps = [ |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:init_ops", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:standard_ops", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python/keras:engine", |
| ], |
| ) |
| |
| py_test( |
| name = "spectral_normalization_test", |
| srcs = ["python/features/python/spectral_normalization_test.py"], |
| python_version = "PY2", |
| srcs_version = "PY2AND3", |
| deps = [ |
| ":spectral_normalization", |
| "//tensorflow/contrib/layers:layers_py", |
| "//tensorflow/contrib/slim", |
| "//tensorflow/python:array_ops", |
| "//tensorflow/python:client_testlib", |
| "//tensorflow/python:dtypes", |
| "//tensorflow/python:framework_ops", |
| "//tensorflow/python:init_ops", |
| "//tensorflow/python:layers", |
| "//tensorflow/python:linalg_ops", |
| "//tensorflow/python:math_ops", |
| "//tensorflow/python:variable_scope", |
| "//tensorflow/python:variables", |
| "//tensorflow/python/keras:layers", |
| "//third_party/py/numpy", |
| ], |
| ) |