diff options
| author |  Saurabh Saxena <srbs@google.com> | 2018-08-30 12:13:54 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-30 12:17:57 -0700 |
| commit | f9bd3bc45a57d218c7fe3e970c4f953da8136dc6 (patch) | |
| tree | b488a89c5f066d2622334be1a7bcb0675d1b8759 /tensorflow/core/BUILD | |
| parent | c44566cdd632e4a9b030244cbc36965cb0ee21c0 (diff) | |
Add lowering pass for functional While op.
This will allow the functional tf.while_loop proposed in https://github.com/tensorflow/community/pull/13 to achieve feature parity with the current implementation.
Lowering is performed only when the "_lower_using_switch_merge" attr is set to True.
PiperOrigin-RevId: 210956432
Diffstat (limited to 'tensorflow/core/BUILD')
| -rw-r--r-- | tensorflow/core/BUILD | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index 608b08efba..9b7a5018b2 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -2742,6 +2742,7 @@ CORE_CPU_LIB_HEADERS = CORE_CPU_BASE_HDRS + [ "common_runtime/graph_optimizer.h", "common_runtime/local_device.h", "common_runtime/lower_if_op.h", + "common_runtime/lower_while_op.h", "common_runtime/memory_types.h", "common_runtime/mkl_cpu_allocator.h", "common_runtime/optimization_registry.h", @@ -2798,6 +2799,7 @@ tf_cuda_library( "common_runtime/hierarchical_tree_broadcaster.cc", "common_runtime/local_device.cc", "common_runtime/lower_if_op.cc", + "common_runtime/lower_while_op.cc", "common_runtime/memory_types.cc", "common_runtime/mkl_cpu_allocator.cc", "common_runtime/optimization_registry.cc", @@ -4601,6 +4603,29 @@ tf_cc_tests( ], ) +tf_cc_tests( + name = "common_runtime_lower_while_op_test", + size = "small", + srcs = ["common_runtime/lower_while_op_test.cc"], + deps = [ + ":all_kernels", + ":core_cpu", + ":core_cpu_internal", + ":direct_session", + ":framework", + ":framework_internal", + ":lib", + ":test", + ":test_main", + ":testlib", + "//tensorflow/cc:cc_ops", + "//tensorflow/cc:cc_ops_internal", + "//tensorflow/cc:client_session", + "//tensorflow/cc:function_ops", + "//tensorflow/cc:ops", + ], +) + # Test data filegroup( name = "image_testdata", |