| 235 | } |
| 236 | |
| 237 | bool TfLiteDriver::Expectation::Check(bool verbose, |
| 238 | const TfLiteTensor& tensor) { |
| 239 | switch (tensor.type) { |
| 240 | case kTfLiteFloat32: |
| 241 | return TypedCheck<float, float>(verbose, tensor); |
| 242 | case kTfLiteInt32: |
| 243 | return TypedCheck<int32_t, float>(verbose, tensor); |
| 244 | case kTfLiteInt64: |
| 245 | return TypedCheck<int64_t, float>(verbose, tensor); |
| 246 | case kTfLiteUInt8: |
| 247 | return TypedCheck<uint8_t, float>(verbose, tensor); |
| 248 | case kTfLiteInt8: |
| 249 | return TypedCheck<int8_t, float>(verbose, tensor); |
| 250 | case kTfLiteBool: |
| 251 | return TypedCheck<bool, float>(verbose, tensor); |
| 252 | case kTfLiteString: |
| 253 | return TypedCheckString(verbose, tensor); |
| 254 | case kTfLiteComplex64: |
| 255 | return TypedCheck<std::complex<float>, std::complex<float>>(verbose, |
| 256 | tensor); |
| 257 | default: |
| 258 | fprintf(stderr, "Unsupported type %d in Check\n", tensor.type); |
| 259 | return false; |
| 260 | } |
| 261 | } |
| 262 | |
| 263 | TfLiteDriver::TfLiteDriver(DelegateType delegate_type, bool reference_kernel) |
| 264 | : delegate_(nullptr, nullptr), |