huajsj commented on code in PR #11183:
URL: https://github.com/apache/tvm/pull/11183#discussion_r865210575
##########
python/tvm/relay/frontend/tflite.py:
##########
@@ -2710,6 +2743,145 @@ def convert_unpack(self, op):
return squeezed
+ def convert_unidirectional_sequence_lstm(self, op):
+ """Long Short Term Memory for TFLite implementation."""
+ if self.is_quantized(op):
+ raise tvm.error.OpNotImplemented(
+ "TFlite quantized UNIDIRECTIONALSEQUENCELSTM operator is not
supported yet."
+ )
+
+ input_tensors = self.get_input_tensors(op)
+ assert len(input_tensors) >= 2, "input tensors length should be >= 2"
Review Comment:
There are 20 tensor get used
`assert len(input_tensors) >= 20, "input tensors length should be >= 20` ?
##########
tests/python/frontend/tflite/test_forward.py:
##########
@@ -4572,6 +4572,36 @@ def test_forward_tflite_float16():
tvm.testing.assert_allclose(tvm_sorted_labels, tflite_sorted_labels)
+#######################################################################
+# Unidirectional Sequence LSTM
+# ---------------------
+def test_unidirectional_sequence_lstm():
Review Comment:
add this function into "_main_" function ?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
