blob: bab37e8906e5c648acdc1556da7e5f4601776ff5 [file] [log] [blame]
Martin Wicke914c8702016-01-25 14:42:53 -08001# Description:
2# contains parts of TensorFlow that are experimental or unstable and which are not supported.
3
4licenses(["notice"]) # Apache 2.0
5
Martin Wicke914c8702016-01-25 14:42:53 -08006package(default_visibility = ["//tensorflow:__subpackages__"])
7
Shanqing Caie2e3a942017-09-25 19:35:53 -07008load("//third_party/mpi:mpi.bzl", "if_mpi")
A. Unique TensorFlower0bb22b82018-01-12 06:15:05 -08009load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")
Yifei Fengdce9a492018-02-22 14:24:57 -080010load("@local_config_tensorrt//:build_defs.bzl", "if_tensorrt")
Shanqing Caie2e3a942017-09-25 19:35:53 -070011
Martin Wicke914c8702016-01-25 14:42:53 -080012py_library(
13 name = "contrib_py",
Dandelion Man?90e42f32017-12-15 18:15:07 -080014 srcs = glob(
15 ["**/*.py"],
16 exclude = [
17 "**/*_test.py",
18 ],
19 ),
Martin Wicke914c8702016-01-25 14:42:53 -080020 srcs_version = "PY2AND3",
A. Unique TensorFlowerf6e72052016-03-23 13:13:09 -080021 visibility = ["//visibility:public"],
A. Unique TensorFlower1e141832016-02-02 09:49:05 -080022 deps = [
A. Unique TensorFlower21f37282017-09-21 12:27:09 -070023 "//tensorflow/contrib/all_reduce",
Christopher Olstonf7f16be2017-04-27 10:21:51 -080024 "//tensorflow/contrib/batching:batch_py",
Eugene Brevdoc3af0832016-04-20 13:38:11 -080025 "//tensorflow/contrib/bayesflow:bayesflow_py",
A. Unique TensorFlowerf92eef72017-06-21 08:53:11 -070026 "//tensorflow/contrib/boosted_trees:init_py",
A. Unique TensorFlower1b881b72017-03-13 16:58:18 -080027 "//tensorflow/contrib/cloud:cloud_py",
A. Unique TensorFlower76989a12018-01-31 09:40:48 -080028 "//tensorflow/contrib/cluster_resolver:cluster_resolver_pip",
Frank Chen3a2971b2017-06-07 22:38:41 -070029 "//tensorflow/contrib/cluster_resolver:cluster_resolver_py",
Sung Jin Hwangca8df262018-01-17 17:05:21 -080030 "//tensorflow/contrib/coder:coder_ops_py",
A. Unique TensorFlowera5e51222016-12-02 09:45:06 -080031 "//tensorflow/contrib/compiler:compiler_py",
A. Unique TensorFlower8bf6ef12016-05-05 08:36:05 -080032 "//tensorflow/contrib/copy_graph:copy_graph_py",
A. Unique TensorFlower34f7db22016-09-22 12:34:47 -080033 "//tensorflow/contrib/crf:crf_py",
Xiaoqiang Zheng962dafe2016-08-26 21:14:00 -080034 "//tensorflow/contrib/cudnn_rnn:cudnn_rnn_py",
Derek Murray866bee32017-05-16 20:10:24 -070035 "//tensorflow/contrib/data",
Dan Manécce0d122016-11-22 17:17:34 -080036 "//tensorflow/contrib/deprecated:deprecated_py",
Eugene Brevdo06b08132016-02-26 17:19:02 -080037 "//tensorflow/contrib/distributions:distributions_py",
Alexandre Passos51441302017-08-18 12:24:39 -070038 "//tensorflow/contrib/eager/python:tfe",
Mustafa Ispir384337f2017-09-06 16:14:41 -070039 "//tensorflow/contrib/estimator:estimator_py",
A. Unique TensorFlower23fdab72016-07-01 17:05:28 -080040 "//tensorflow/contrib/factorization:factorization_py",
A. Unique TensorFloweraf46e2c2018-02-06 15:09:51 -080041 "//tensorflow/contrib/feature_column:feature_column_py",
A. Unique TensorFlowerce575f12016-04-29 09:08:46 -080042 "//tensorflow/contrib/ffmpeg:ffmpeg_ops_py",
A. Unique TensorFlower5b5b8412016-03-24 08:19:12 -080043 "//tensorflow/contrib/framework:framework_py",
Yangzihao Wang4bbd9bd2017-07-20 09:32:48 -070044 "//tensorflow/contrib/fused_conv:fused_conv_py",
A. Unique TensorFlower19680e62017-08-31 15:00:04 -070045 "//tensorflow/contrib/gan",
A. Unique TensorFlower44fcfbd2016-06-29 06:54:02 -080046 "//tensorflow/contrib/graph_editor:graph_editor_py",
Illia Polosukhin5c9bc512016-04-18 17:56:51 -080047 "//tensorflow/contrib/grid_rnn:grid_rnn_py",
Vijay Vasudevan93a975e2017-02-17 17:05:49 -080048 "//tensorflow/contrib/hooks",
A. Unique TensorFlowerbc68dc82017-09-15 15:28:15 -070049 "//tensorflow/contrib/image:distort_image_py",
Dan Ringwalte993a932016-12-07 14:20:21 -080050 "//tensorflow/contrib/image:image_py",
Patrick Nguyen69a4cf82017-04-17 20:41:44 -080051 "//tensorflow/contrib/image:single_image_random_dot_stereograms_py",
Rohan Jainb1472252016-12-02 18:26:44 -080052 "//tensorflow/contrib/input_pipeline:input_pipeline_py",
Stephan Hoyer65b16772016-11-02 10:48:36 -080053 "//tensorflow/contrib/integrate:integrate_py",
Michael Cased90054e2018-02-07 14:36:00 -080054 "//tensorflow/contrib/kafka",
Francois Cholletf49f8012017-03-15 12:53:51 -080055 "//tensorflow/contrib/keras",
Gunhan Gulsoy24179a12017-03-27 16:13:33 -080056 "//tensorflow/contrib/kernel_methods",
A. Unique TensorFlowerac138362017-09-27 18:52:01 -070057 "//tensorflow/contrib/kfac",
Stephan Hoyer9d20f4e2016-11-14 17:24:44 -080058 "//tensorflow/contrib/labeled_tensor",
A. Unique TensorFlower1e141832016-02-02 09:49:05 -080059 "//tensorflow/contrib/layers:layers_py",
Illia Polosukhin5c9bc512016-04-18 17:56:51 -080060 "//tensorflow/contrib/learn",
Jianwei Xie7ed760f2016-12-03 09:31:42 -080061 "//tensorflow/contrib/legacy_seq2seq:seq2seq_py",
A. Unique TensorFlowere4532d22017-12-22 12:42:59 -080062 "//tensorflow/contrib/libsvm",
Ian Langmore88165b42016-11-21 20:54:07 -080063 "//tensorflow/contrib/linalg:linalg_py",
Petros Molf5398a32017-03-16 10:02:12 -080064 "//tensorflow/contrib/linear_optimizer:sdca_estimator_py",
A. Unique TensorFlower94a992c2016-02-23 18:51:19 -080065 "//tensorflow/contrib/linear_optimizer:sdca_ops_py",
Andrew Selle0b154392017-11-10 10:35:35 -080066 "//tensorflow/contrib/lite/python:lite",
Yutaka Leone06a4a22016-03-15 18:00:02 -080067 "//tensorflow/contrib/lookup:lookup_py",
A. Unique TensorFlower5b5b8412016-03-24 08:19:12 -080068 "//tensorflow/contrib/losses:losses_py",
A. Unique TensorFlowerdeb72df2017-10-12 12:41:40 -070069 "//tensorflow/contrib/losses:metric_learning_py",
Jingyue Wuccf9a752017-02-28 22:09:13 -080070 "//tensorflow/contrib/memory_stats:memory_stats_py",
David Soergel34c49b12017-07-28 09:00:28 -070071 "//tensorflow/contrib/meta_graph_transform",
A. Unique TensorFlowerc5073de2016-04-07 15:55:59 -080072 "//tensorflow/contrib/metrics:metrics_py",
A. Unique TensorFlower7ece1c02017-11-01 11:55:32 -070073 "//tensorflow/contrib/model_pruning",
A. Unique TensorFlower08a3e362017-03-31 11:26:42 -080074 "//tensorflow/contrib/nccl:nccl_py",
A. Unique TensorFlower337b12f2017-08-17 00:16:28 -070075 "//tensorflow/contrib/nearest_neighbor:nearest_neighbor_py",
Martin Wickeb61c1422016-12-28 11:50:15 -080076 "//tensorflow/contrib/nn:nn_py",
A. Unique TensorFlowerb6259682016-05-17 14:31:29 -080077 "//tensorflow/contrib/opt:opt_py",
Shanqing Caife840612017-12-06 18:43:24 -080078 "//tensorflow/contrib/periodic_resample:init_py",
A. Unique TensorFlower576c7b12017-07-12 09:01:22 -070079 "//tensorflow/contrib/predictor",
Pete Wardenca4e0532016-04-22 17:30:38 -080080 "//tensorflow/contrib/quantization:quantization_py",
Suharsh Sivakumar955c5252017-10-02 23:04:00 -070081 "//tensorflow/contrib/quantize:quantize_graph",
A. Unique TensorFlowerb9706522018-01-30 11:09:06 -080082 "//tensorflow/contrib/py2tf",
A. Unique TensorFlower9698e9d2018-01-11 12:12:06 -080083 "//tensorflow/contrib/receptive_field:receptive_field_py",
Andrew Harp6e3e7d12017-08-21 12:10:44 -070084 "//tensorflow/contrib/reduce_slice_ops:reduce_slice_ops_py",
A. Unique TensorFlower8f2bf532017-06-16 11:37:40 -070085 "//tensorflow/contrib/remote_fused_graph/pylib:remote_fused_graph_ops_py",
A. Unique TensorFlowera80c8b52017-06-26 05:47:31 -070086 "//tensorflow/contrib/resampler:resampler_py",
A. Unique TensorFlowerd1a31022016-04-20 06:29:39 -080087 "//tensorflow/contrib/rnn:rnn_py",
Sukriti Rameshdd64c322017-03-09 18:59:30 -080088 "//tensorflow/contrib/saved_model:saved_model_py",
Patrick Nguyenc5ab3dd2016-10-20 12:09:18 -080089 "//tensorflow/contrib/seq2seq:seq2seq_py",
Jonathan Hseu1b5235f2017-06-09 10:37:18 -070090 "//tensorflow/contrib/signal:signal_py",
A. Unique TensorFlower65ba9e52016-06-04 19:17:37 -080091 "//tensorflow/contrib/slim",
Nathan Silbermane70b7552016-08-17 13:27:56 -080092 "//tensorflow/contrib/slim:nets",
A. Unique TensorFlower4bab2622016-11-14 14:37:20 -080093 "//tensorflow/contrib/solvers:solvers_py",
Benoit Steiner639b4e72017-02-08 09:25:09 -080094 "//tensorflow/contrib/sparsemax:sparsemax_py",
A. Unique TensorFlower65e31292016-10-20 16:30:37 -080095 "//tensorflow/contrib/specs",
A. Unique TensorFlower90733a82017-04-20 11:03:06 -080096 "//tensorflow/contrib/staging",
Andrew Harp1743ad82016-11-11 16:43:51 -080097 "//tensorflow/contrib/stat_summarizer:stat_summarizer_py",
Geoffrey Irvingcc454562017-04-17 13:30:47 -080098 "//tensorflow/contrib/stateless",
Alexandre Passosff7b9a62017-10-25 18:46:35 -070099 "//tensorflow/contrib/summary:summary",
Justine Tunneye1216672016-12-29 22:46:24 -0800100 "//tensorflow/contrib/tensor_forest:init_py",
Dan Smilkov9e33df52016-09-21 18:48:55 -0800101 "//tensorflow/contrib/tensorboard",
A. Unique TensorFlowerd1aed652016-02-25 10:39:17 -0800102 "//tensorflow/contrib/testing:testing_py",
Wei Ho458f94c2017-05-31 17:07:05 -0700103 "//tensorflow/contrib/text:text_py",
A. Unique TensorFlower113093b2016-09-30 12:20:29 -0800104 "//tensorflow/contrib/tfprof",
Allen Lavoie1b7d1312017-07-05 15:31:54 -0700105 "//tensorflow/contrib/timeseries",
A. Unique TensorFlowera0544b02017-08-18 14:30:07 -0700106 "//tensorflow/contrib/tpu",
A. Unique TensorFlower966016b2017-10-26 17:16:35 -0700107 "//tensorflow/contrib/tpu:tpu_py",
Eugene Brevdo955efc92016-08-16 14:20:09 -0800108 "//tensorflow/contrib/training:training_py",
Geoffrey Irvingd7970482016-02-08 09:34:09 -0800109 "//tensorflow/contrib/util:util_py",
A. Unique TensorFlower966016b2017-10-26 17:16:35 -0700110 "//tensorflow/python:util",
Yifei Fengdce9a492018-02-22 14:24:57 -0800111 ] + if_mpi(["//tensorflow/contrib/mpi_collectives:mpi_collectives_py"]) + if_tensorrt([
112 "//tensorflow/contrib/tensorrt:init_py",
113 ]),
Martin Wicke914c8702016-01-25 14:42:53 -0800114)
115
A. Unique TensorFloweraaca1cc2016-03-17 22:41:12 -0800116cc_library(
A. Unique TensorFlowereed87c92016-03-22 15:13:22 -0800117 name = "contrib_kernels",
A. Unique TensorFloweraaca1cc2016-03-17 22:41:12 -0800118 visibility = ["//visibility:public"],
119 deps = [
A. Unique TensorFlowerf92eef72017-06-21 08:53:11 -0700120 "//tensorflow/contrib/boosted_trees:boosted_trees_kernels",
Sung Jin Hwangca8df262018-01-17 17:05:21 -0800121 "//tensorflow/contrib/coder:all_kernels",
A. Unique TensorFlower9f8d8bc2017-08-03 13:19:13 -0700122 "//tensorflow/contrib/cudnn_rnn:cudnn_rnn_kernels",
Derek Murray7e3acec2018-02-07 20:17:11 -0800123 "//tensorflow/contrib/data/kernels:dataset_kernels",
A. Unique TensorFlower23fdab72016-07-01 17:05:28 -0800124 "//tensorflow/contrib/factorization/kernels:all_kernels",
Rohan Jainb1472252016-12-02 18:26:44 -0800125 "//tensorflow/contrib/input_pipeline:input_pipeline_ops_kernels",
A. Unique TensorFlower392e2ed2016-05-18 14:19:55 -0800126 "//tensorflow/contrib/layers:sparse_feature_cross_op_kernel",
A. Unique TensorFlower337b12f2017-08-17 00:16:28 -0700127 "//tensorflow/contrib/nearest_neighbor:nearest_neighbor_ops_kernels",
Adam Roberts6285db22017-10-11 16:09:14 -0700128 "//tensorflow/contrib/rnn:all_kernels",
A. Unique TensorFlower4f3ae762017-06-01 15:52:27 -0700129 "//tensorflow/contrib/seq2seq:beam_search_ops_kernels",
A. Unique TensorFlowerf5e500c2017-06-22 11:29:54 -0700130 "//tensorflow/contrib/tensor_forest:model_ops_kernels",
131 "//tensorflow/contrib/tensor_forest:stats_ops_kernels",
Kiril Gorovoyc5cd1f72017-03-28 16:54:38 -0800132 "//tensorflow/contrib/tensor_forest:tensor_forest_kernels",
Wei Ho458f94c2017-05-31 17:07:05 -0700133 "//tensorflow/contrib/text:all_kernels",
A. Unique TensorFlower0bb22b82018-01-12 06:15:05 -0800134 ] + if_mpi(["//tensorflow/contrib/mpi_collectives:mpi_collectives_py"]) + if_cuda([
135 "//tensorflow/contrib/nccl:nccl_kernels",
136 ]),
A. Unique TensorFloweraaca1cc2016-03-17 22:41:12 -0800137)
138
A. Unique TensorFlowerc25a3fa2016-04-13 19:54:30 -0800139cc_library(
140 name = "contrib_ops_op_lib",
141 visibility = ["//visibility:public"],
142 deps = [
A. Unique TensorFlowerf92eef72017-06-21 08:53:11 -0700143 "//tensorflow/contrib/boosted_trees:boosted_trees_ops_op_lib",
Sung Jin Hwangca8df262018-01-17 17:05:21 -0800144 "//tensorflow/contrib/coder:all_ops",
A. Unique TensorFlower9f8d8bc2017-08-03 13:19:13 -0700145 "//tensorflow/contrib/cudnn_rnn:cudnn_rnn_ops_op_lib",
Derek Murray7e3acec2018-02-07 20:17:11 -0800146 "//tensorflow/contrib/data:dataset_ops_op_lib",
A. Unique TensorFlower23fdab72016-07-01 17:05:28 -0800147 "//tensorflow/contrib/factorization:all_ops",
Jonathan Hseu1283b842016-09-29 15:05:32 -0800148 "//tensorflow/contrib/framework:all_ops",
Rohan Jainb1472252016-12-02 18:26:44 -0800149 "//tensorflow/contrib/input_pipeline:input_pipeline_ops_op_lib",
Michael Cased90054e2018-02-07 14:36:00 -0800150 "//tensorflow/contrib/kafka:kafka_ops_op_lib",
A. Unique TensorFlower392e2ed2016-05-18 14:19:55 -0800151 "//tensorflow/contrib/layers:sparse_feature_cross_op_op_lib",
A. Unique TensorFlowercf574652017-04-28 17:13:26 -0800152 "//tensorflow/contrib/nccl:nccl_ops_op_lib",
A. Unique TensorFlower337b12f2017-08-17 00:16:28 -0700153 "//tensorflow/contrib/nearest_neighbor:nearest_neighbor_ops_op_lib",
Adam Roberts6285db22017-10-11 16:09:14 -0700154 "//tensorflow/contrib/rnn:all_ops",
A. Unique TensorFlower4f3ae762017-06-01 15:52:27 -0700155 "//tensorflow/contrib/seq2seq:beam_search_ops_op_lib",
A. Unique TensorFlowerf5e500c2017-06-22 11:29:54 -0700156 "//tensorflow/contrib/tensor_forest:model_ops_op_lib",
157 "//tensorflow/contrib/tensor_forest:stats_ops_op_lib",
Kiril Gorovoyc5cd1f72017-03-28 16:54:38 -0800158 "//tensorflow/contrib/tensor_forest:tensor_forest_ops_op_lib",
Wei Ho458f94c2017-05-31 17:07:05 -0700159 "//tensorflow/contrib/text:all_ops",
Frank Chena4660cc2017-06-20 17:11:19 -0700160 "//tensorflow/contrib/tpu:all_ops",
A. Unique TensorFlowerc25a3fa2016-04-13 19:54:30 -0800161 ],
162)
163
Martin Wicke914c8702016-01-25 14:42:53 -0800164filegroup(
165 name = "all_files",
166 srcs = glob(
167 ["**/*"],
168 exclude = [
169 "**/METADATA",
170 "**/OWNERS",
171 ],
172 ),
173 visibility = ["//tensorflow:__subpackages__"],
174)