| 65 | } |
| 66 | |
| 67 | armnnSerializer::DataType GetFlatBufferDataType(armnn::DataType dataType) |
| 68 | { |
| 69 | switch (dataType) |
| 70 | { |
| 71 | case armnn::DataType::Float32: |
| 72 | return armnnSerializer::DataType::DataType_Float32; |
| 73 | case armnn::DataType::Float16: |
| 74 | return armnnSerializer::DataType::DataType_Float16; |
| 75 | case armnn::DataType::Signed32: |
| 76 | return armnnSerializer::DataType::DataType_Signed32; |
| 77 | case armnn::DataType::Signed64: |
| 78 | return armnnSerializer::DataType::DataType_Signed64; |
| 79 | case armnn::DataType::QSymmS16: |
| 80 | return armnnSerializer::DataType::DataType_QSymmS16; |
| 81 | case armnn::DataType::QAsymmS8: |
| 82 | return armnnSerializer::DataType::DataType_QAsymmS8; |
| 83 | case armnn::DataType::QAsymmU8: |
| 84 | return armnnSerializer::DataType::DataType_QAsymmU8; |
| 85 | case armnn::DataType::QSymmS8: |
| 86 | return armnnSerializer::DataType::DataType_QSymmS8; |
| 87 | case armnn::DataType::Boolean: |
| 88 | return armnnSerializer::DataType::DataType_Boolean; |
| 89 | default: |
| 90 | return armnnSerializer::DataType::DataType_Float16; |
| 91 | } |
| 92 | } |
| 93 | |
| 94 | armnnSerializer::DataLayout GetFlatBufferDataLayout(armnn::DataLayout dataLayout) |
| 95 | { |