diff options
| author |  Saurabh Saxena <srbs@google.com> | 2018-09-18 20:33:07 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-18 20:36:44 -0700 |
| commit | 1b2d0fcee82ec501cc692dc735065d73c6b5b834 (patch) | |
| tree | 2b29182792e193e9695b796d7e4c92b95ab040c6 /tensorflow/python/BUILD | |
| parent | 9fe177881224571aff0c267593f747f5fd7a2967 (diff) | |
First commit for functional while loop.
Supports single and double derivatives but does not supporting nesting yet.
https://github.com/tensorflow/community/pull/13
PiperOrigin-RevId: 213565971
Diffstat (limited to 'tensorflow/python/BUILD')
| -rw-r--r-- | tensorflow/python/BUILD | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 2eeae773d3..d70e9c5798 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -1998,6 +1998,29 @@ py_library( ) py_library( + name = "while_v2", + srcs = [ + "ops/while_v2.py", + ], + srcs_version = "PY2AND3", + deps = [ + ":array_ops", + ":cond_v2_impl", + ":constant_op", + ":control_flow_util", + ":framework_ops", + ":function_def_to_graph", + ":functional_ops_gen", + ":gradients_impl", + ":list_ops", + ":tensor_shape", + ":util", + "//tensorflow/core:protos_all_py", + "//tensorflow/python/eager:function", + ], +) + +py_library( name = "cond_v2_impl", srcs = [ "ops/cond_v2_impl.py", |