diff options
Diffstat (limited to 'tensorflow/contrib/lite/toco/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/toco/BUILD | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/toco/BUILD b/tensorflow/contrib/lite/toco/BUILD index bea90f1ce8..96b88b60fc 100644 --- a/tensorflow/contrib/lite/toco/BUILD +++ b/tensorflow/contrib/lite/toco/BUILD @@ -347,6 +347,7 @@ tf_cc_test( "//tensorflow/core:framework", "//tensorflow/core:graph", "//tensorflow/core:lib", + "//tensorflow/core:ops", "//tensorflow/core:protos_all_cc", "@com_google_googletest//:gtest_main", ], @@ -407,8 +408,11 @@ tf_cc_binary( ":toco_port", ":toco_tooling", ":types_proto_cc", - "//tensorflow/core:lib", "@com_google_absl//absl/strings", + "//tensorflow/core:lib", + # We cannot embed the core:ops dependency directly into :toco_tooling as + # it can conflict with downstream deps when toco is used as a library. + "//tensorflow/core:ops", ], ) |