MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / IsBasicLSTMOp

Function IsBasicLSTMOp

tensorflow/compiler/mlir/lite/flatbuffer_import.cc:532–538  ·  view source on GitHub ↗

Returns true if this is a basic LSTM op.

Source from the content-addressed store, hash-verified

530
531// Returns true if this is a basic LSTM op.
532bool 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.
541bool IsCustomOp(const std::string& op_name) {

Callers 1

ConvertOpFunction · 0.85

Calls 1

AsLSTMOptionsMethod · 0.80

Tested by

no test coverage detected