Additional toco options besides input, output, shape.
| 138 | |
| 139 | |
| 140 | class ExtraTocoOptions(object): |
| 141 | """Additional toco options besides input, output, shape.""" |
| 142 | |
| 143 | def __init__(self): |
| 144 | # Whether to ignore control dependency nodes. |
| 145 | self.drop_control_dependency = False |
| 146 | # Allow custom ops in the toco conversion. |
| 147 | self.allow_custom_ops = False |
| 148 | # Rnn states that are used to support rnn / lstm cells. |
| 149 | self.rnn_states = None |
| 150 | # Split the LSTM inputs from 5 inoputs to 18 inputs for TFLite. |
| 151 | self.split_tflite_lstm_inputs = None |
| 152 | # The inference input type passed to TFLiteConvert. |
| 153 | self.inference_input_type = None |
| 154 | # The inference output type passed to TFLiteConvert. |
| 155 | self.inference_output_type = None |
| 156 | |
| 157 | |
| 158 |
no outgoing calls
no test coverage detected