| 202 | } |
| 203 | |
| 204 | int GetVersion(const OperatorSignature& op_signature) const override { |
| 205 | const string& input_name = op_signature.op->inputs[0]; |
| 206 | const Array& input_array = op_signature.model->GetArray(input_name); |
| 207 | // Version 2 supports signed int8 input types. |
| 208 | if (input_array.data_type == ArrayDataType::kInt8) { |
| 209 | return 2; |
| 210 | } |
| 211 | return 1; |
| 212 | } |
| 213 | }; |
| 214 | |
| 215 | class AddN : public BuiltinOperator<AddNOperator, ::tflite::AddNOptions, |
no outgoing calls