File tree Expand file tree Collapse file tree 1 file changed +14
-8
lines changed Expand file tree Collapse file tree 1 file changed +14
-8
lines changed Original file line number Diff line number Diff line change @@ -135,24 +135,30 @@ auto TensorImpl::getNumBytes() -> size_t {
135
135
const auto type = getType ();
136
136
137
137
switch (type) {
138
- case TensorType::FLOAT32:
139
- numBytes = sizeof (float );
140
- break ;
141
138
case TensorType::FLOAT16:
142
139
numBytes = 2 ;
143
140
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 );
149
143
break ;
150
144
case TensorType::INT8:
151
145
numBytes = sizeof (int8_t );
152
146
break ;
147
+ case TensorType::INT16:
148
+ numBytes = sizeof (int16_t );
149
+ break ;
150
+ case TensorType::INT32:
151
+ numBytes = sizeof (int32_t );
152
+ break ;
153
153
case TensorType::UINT8:
154
154
numBytes = sizeof (uint8_t );
155
155
break ;
156
+ case TensorType::UINT16:
157
+ numBytes = sizeof (uint16_t );
158
+ break ;
159
+ case TensorType::UINT32:
160
+ numBytes = sizeof (uint32_t );
161
+ break ;
156
162
default :
157
163
return {};
158
164
}
You can’t perform that action at this time.
0 commit comments