def infer(node: Node): # there are limitations coming from ONNX LSTM definition and normalization rules assert len(node.in_nodes()) >= 3 # X, W and R assert len(node.in_nodes()) <= 7 assert len(node.out_nodes()) <= 3 rnn_infer(node, [1, 2])
def infer(node: Node): assert len(node.in_nodes()) >= 3 # X, W and R assert len(node.in_nodes()) <= 5 assert len(node.out_nodes()) <= 2 rnn_infer(node, [1])