Add BroadcastTo to whitelisted Flex ops list.

PiperOrigin-RevId: 277206853 Change-Id: I39872347e9f74b8f5079cc0ee967e864efeea06a
2019-10-28 21:13:44 -07:00 · 2019-10-28 21:13:44 -07:00 · f0570083f5
commit f0570083f5
parent a96887c483
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
+++ b/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
@ -68,6 +68,7 @@ bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) {
          "BiasAddV1",
          "BroadcastArgs",
          "BroadcastGradientArgs",
+          "BroadcastTo",
          "Cast",
          "Ceil",
          "CheckNumerics",