diff options
Diffstat (limited to 'tensorflow/core/grappler/BUILD')
-rw-r--r-- | tensorflow/core/grappler/BUILD | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/BUILD b/tensorflow/core/grappler/BUILD index 9dcc6765f5..7c6fe56e1f 100644 --- a/tensorflow/core/grappler/BUILD +++ b/tensorflow/core/grappler/BUILD @@ -33,6 +33,7 @@ tf_cc_test( name = "utils_test", srcs = ["utils_test.cc"], deps = [ + ":grappler_item", ":utils", "//tensorflow/cc:cc_ops", "//tensorflow/core:all_kernels", @@ -151,3 +152,32 @@ tf_cc_test( "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder", ], ) + +cc_library( + name = "mutable_graph_view", + srcs = [ + "mutable_graph_view.cc", + ], + hdrs = ["mutable_graph_view.h"], + visibility = ["//visibility:public"], + deps = [ + ":graph_view", + ":grappler_item", + ":utils", + "//tensorflow/core:lib", + "//tensorflow/core:protos_all_cc", + ], +) + +tf_cc_test( + name = "mutable_graph_view_test", + srcs = ["mutable_graph_view_test.cc"], + deps = [ + ":grappler_item", + ":mutable_graph_view", + "//tensorflow/cc:cc_ops", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder", + ], +) |