We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8f8f515 commit 12dcb4eCopy full SHA for 12dcb4e
1 file changed
src/webnn/native/nnapi/GraphNnapi.h
100644
100755
@@ -144,10 +144,14 @@ namespace webnn::native::nnapi {
144
const void* buffer = nullptr) {
145
std::shared_ptr<NodeInfo> node = std::make_shared<NodeInfo>();
146
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]));
+ if (desc->dimensionsCount == 0) {
+ 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
+ }
153
}
-
154
+
155
MaybeError error;
156
if (buffer) {
157
error = mNnapiMgr->CreateOperandAndSetMemory(name, node, buffer);
0 commit comments