ONE - On-device Neural Engine
Loading...
Searching...
No Matches
onnx_legalizer.LegalizeOptions Class Reference

Static Public Attributes

bool unroll_rnn = False
 
bool unroll_lstm = False
 

Detailed Description

Controls transformations that legalizer apply

Attributes:
    unroll_rnn (bool): default is False. If True - unrolls RNN operations
    unroll_lstm (bool): default is False. If True - unrolls LSTM operations

Definition at line 36 of file onnx_legalizer.py.

Field Documentation

◆ unroll_lstm

bool onnx_legalizer.LegalizeOptions.unroll_lstm = False
static

Definition at line 45 of file onnx_legalizer.py.

◆ unroll_rnn

bool onnx_legalizer.LegalizeOptions.unroll_rnn = False
static

Definition at line 44 of file onnx_legalizer.py.


The documentation for this class was generated from the following file: