| 13 | using namespace lite; |
| 14 | |
| 15 | size_t Layout::get_elem_size() const { |
| 16 | size_t elesize = 1; |
| 17 | switch (data_type) { |
| 18 | case LiteDataType::LITE_INT64: |
| 19 | elesize = 8; |
| 20 | break; |
| 21 | case LiteDataType::LITE_FLOAT: |
| 22 | case LiteDataType::LITE_INT: |
| 23 | case LiteDataType::LITE_UINT: |
| 24 | elesize = 4; |
| 25 | break; |
| 26 | case LiteDataType::LITE_HALF: |
| 27 | case LiteDataType::LITE_INT16: |
| 28 | case LiteDataType::LITE_UINT16: |
| 29 | elesize = 2; |
| 30 | break; |
| 31 | case LiteDataType::LITE_INT8: |
| 32 | case LiteDataType::LITE_UINT8: |
| 33 | elesize = 1; |
| 34 | break; |
| 35 | default: |
| 36 | LITE_THROW("not support data type."); |
| 37 | } |
| 38 | return elesize; |
| 39 | } |
| 40 | |
| 41 | bool Layout::operator==(const Layout& other) const { |
| 42 | bool equal = true; |
no outgoing calls