Skip to content

Commit 6197881

Browse files
committed
Support more QNN tensor types
1 parent 6bf9df0 commit 6197881

File tree

1 file changed

+14
-8
lines changed

1 file changed

+14
-8
lines changed

source/qnn/tensor.cpp

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -135,24 +135,30 @@ auto TensorImpl::getNumBytes() -> size_t {
135135
const auto type = getType();
136136

137137
switch (type) {
138-
case TensorType::FLOAT32:
139-
numBytes = sizeof(float);
140-
break;
141138
case TensorType::FLOAT16:
142139
numBytes = 2;
143140
break;
144-
case TensorType::INT32:
145-
numBytes = sizeof(int32_t);
146-
break;
147-
case TensorType::UINT32:
148-
numBytes = sizeof(uint32_t);
141+
case TensorType::FLOAT32:
142+
numBytes = sizeof(float);
149143
break;
150144
case TensorType::INT8:
151145
numBytes = sizeof(int8_t);
152146
break;
147+
case TensorType::INT16:
148+
numBytes = sizeof(int16_t);
149+
break;
150+
case TensorType::INT32:
151+
numBytes = sizeof(int32_t);
152+
break;
153153
case TensorType::UINT8:
154154
numBytes = sizeof(uint8_t);
155155
break;
156+
case TensorType::UINT16:
157+
numBytes = sizeof(uint16_t);
158+
break;
159+
case TensorType::UINT32:
160+
numBytes = sizeof(uint32_t);
161+
break;
156162
default:
157163
return {};
158164
}

0 commit comments

Comments
 (0)