Skip to content

Commit 1277e11

Browse files
authored
Merge pull request #278 from vbenni/mobilenetv2_main
(nnapi) Added support for zero input dimension for Gemm op
2 parents d4318ce + 12dcb4e commit 1277e11

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

src/webnn/native/nnapi/GraphNnapi.h

100644100755
Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -144,10 +144,14 @@ namespace webnn::native::nnapi {
144144
const void* buffer = nullptr) {
145145
std::shared_ptr<NodeInfo> node = std::make_shared<NodeInfo>();
146146
node->type = desc->type;
147-
for (size_t i = 0; i < desc->dimensionsCount; i++) {
148-
node->dimensions.push_back(static_cast<uint32_t>(desc->dimensions[i]));
147+
if (desc->dimensionsCount == 0) {
148+
node->dimensions.push_back(static_cast<uint32_t>(1));
149+
} else {
150+
for (size_t i = 0; i < desc->dimensionsCount; i++) {
151+
node->dimensions.push_back(static_cast<uint32_t>(desc->dimensions[i]));
152+
}
149153
}
150-
154+
151155
MaybeError error;
152156
if (buffer) {
153157
error = mNnapiMgr->CreateOperandAndSetMemory(name, node, buffer);

0 commit comments

Comments
 (0)