Returns true if this is a basic LSTM op.
| 530 | |
| 531 | // Returns true if this is a basic LSTM op. |
| 532 | bool IsBasicLSTMOp(tflite::BuiltinOptionsUnion op_union) { |
| 533 | if (const auto* op = op_union.AsLSTMOptions()) { |
| 534 | return op->kernel_type == tflite::LSTMKernelType_BASIC; |
| 535 | } else { |
| 536 | return false; |
| 537 | } |
| 538 | } |
| 539 | |
| 540 | // Returns true if this is a custom op. |
| 541 | bool IsCustomOp(const std::string& op_name) { |
no test coverage detected