Skip to content

Commit

Permalink
Update tvm.js (#1349)
Browse files Browse the repository at this point in the history
  • Loading branch information
lutzroeder committed Sep 14, 2024
1 parent 323458c commit 50c4518
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion source/tvm.js
Original file line number Diff line number Diff line change
Expand Up @@ -178,7 +178,10 @@ tvm.Graph = class {
blocks.get(nodeName).params.push({ name: argumentName, id: key });
values.map(key, null, tensors.get(key));
}
this.nodes = blocks.values().map((block) => new tvm.Node(metadata, block, new Map(), values));
for (const block of blocks.values()) {
const node = new tvm.Node(metadata, block, new Map(), values);
this.nodes.push(node);
}
}
}
};
Expand Down

0 comments on commit 50c4518

Please sign in to comment.