| 304 | } |
| 305 | |
| 306 | void ValidateWeightDataType(const TensorInfo& inputInfo, |
| 307 | const TensorInfo& weightInfo, |
| 308 | const std::string& descName) |
| 309 | { |
| 310 | const DataType inputType = inputInfo.GetDataType(); |
| 311 | if (IsQuantized8BitType(inputType)) |
| 312 | { |
| 313 | const std::vector<DataType> validTypes = |
| 314 | { |
| 315 | DataType::QAsymmS8, |
| 316 | DataType::QAsymmU8, |
| 317 | DataType::QSymmS8 |
| 318 | }; |
| 319 | |
| 320 | ValidateDataTypes(weightInfo, validTypes, descName); |
| 321 | } |
| 322 | else |
| 323 | { |
| 324 | ValidateTensorDataTypesMatch(inputInfo, weightInfo, descName, "input", "weight"); |
| 325 | } |
| 326 | } |
| 327 | |
| 328 | void ValidatePerAxisQuantizationDimension(const TensorInfo& tensorInfo, |
| 329 | const std::string& descName, |
no test coverage detected