Check if a TensorFlow Op is unsupported by the Flex runtime.
| 69 | |
| 70 | // Check if a TensorFlow Op is unsupported by the Flex runtime. |
| 71 | bool IsUnsupportedFlexOp(const string& tensorflow_op) { |
| 72 | if (IsControlFlowOp(tensorflow_op)) { |
| 73 | return true; |
| 74 | } |
| 75 | // `HashTableV2` isn't supported for now since it requires an additional |
| 76 | // initialization step. |
| 77 | // TODO(b/117651199): Support `HashTableV2` with Flex runtime. |
| 78 | if (tensorflow_op == "HashTableV2") { |
| 79 | return true; |
| 80 | } |
| 81 | return false; |
| 82 | } |
| 83 | |
| 84 | // Map from operator name to TF Lite enum value, for all builtins. |
| 85 | const std::map<string, BuiltinOperator>& GetBuiltinOpsMap() { |
no test coverage detected