diff options
Diffstat (limited to 'tensorflow/core/grappler/optimizers/BUILD')
-rw-r--r-- | tensorflow/core/grappler/optimizers/BUILD | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/optimizers/BUILD b/tensorflow/core/grappler/optimizers/BUILD index bd96e2b33c..2ea150ce18 100644 --- a/tensorflow/core/grappler/optimizers/BUILD +++ b/tensorflow/core/grappler/optimizers/BUILD @@ -26,6 +26,40 @@ filegroup( ) cc_library( + name = "auto_parallel", + srcs = ["auto_parallel.cc"], + hdrs = [ + "auto_parallel.h", + ], + visibility = ["//visibility:public"], + deps = [ + ":graph_optimizer", + "//tensorflow/core:framework", + "//tensorflow/core:lib", + "//tensorflow/core:protos_all_cc", + "//tensorflow/core/grappler:devices", + "//tensorflow/core/grappler:grappler_item", + "//tensorflow/core/grappler:utils", + "//tensorflow/core/grappler/clusters:cluster", + ], +) + +cc_test( + name = "auto_parallel_test", + srcs = ["auto_parallel_test.cc"], + deps = [ + ":auto_parallel", + "//tensorflow/cc:cc_ops", + "//tensorflow/core:protos_all_cc", + "//tensorflow/core:test", + "//tensorflow/core:test_main", + "//tensorflow/core/grappler:grappler_item", + "//tensorflow/core/grappler:utils", + "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder", + ], +) + +cc_library( name = "constant_folding", srcs = ["constant_folding.cc"], hdrs = [ @@ -179,6 +213,7 @@ cc_library( ":constant_folding", ":graph_optimizer", ":layout_optimizer", + ":memory_optimizer", ":model_pruner", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", |