quark.onnx.tools.convert_lstm_to_customlstm

Module Contents

Functions

quark.onnx.tools.convert_lstm_to_customlstm.convert_lstm_to_customlstm(model: onnx.ModelProto) → Any

Convert Custom LSTM to LSTM. :param model: source model :return: converted model

quark.onnx.tools.convert_lstm_to_customlstm.custom_ops_infer_shapes(model: onnx.ModelProto) → Any

Generate value info for output tensors of custom ops. :param model: source model :return: converted model