| 216 | |
| 217 | |
| 218 | inline armnn::DataType getArmNNDataType(std::string& descr) |
| 219 | { |
| 220 | if(descr.find("f4") != std::string::npos || descr.find("f8") != std::string::npos) |
| 221 | { |
| 222 | return armnn::DataType::Float32; |
| 223 | } |
| 224 | else if (descr.find("f2") != std::string::npos) |
| 225 | { |
| 226 | return armnn::DataType::Float16; |
| 227 | } |
| 228 | else if (descr.find("i8") != std::string::npos) |
| 229 | { |
| 230 | return armnn::DataType::Signed64; |
| 231 | } |
| 232 | else if (descr.find("i4") != std::string::npos) |
| 233 | { |
| 234 | return armnn::DataType::Signed32; |
| 235 | } |
| 236 | else if (descr.find("i2") != std::string::npos) |
| 237 | { |
| 238 | return armnn::DataType::QSymmS16; |
| 239 | } |
| 240 | else if (descr.find("i1") != std::string::npos) |
| 241 | { |
| 242 | return armnn::DataType::QSymmS8; |
| 243 | } |
| 244 | else if (descr.find("u1") != std::string::npos) |
| 245 | { |
| 246 | return armnn::DataType::QAsymmU8; |
| 247 | } |
| 248 | else |
| 249 | { |
| 250 | throw armnn::Exception(fmt::format("Numpy data type:{} not supported. {}", |
| 251 | descr, CHECK_LOCATION().AsString())); |
| 252 | } |
| 253 | } |
| 254 | |
| 255 | inline std::string getNumpyDescr(armnn::DataType dType) |
| 256 | { |
no test coverage detected