Skip to content

Commit 12dcb4e

Browse files
committed
(nnapi) - Added support for zero input dimension for Gemm op
Signed-Off-by: Vijeetkumar Benni <vijeetkumar.benni@intel.com>
1 parent 8f8f515 commit 12dcb4e

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)