Skip to content

Commit

Permalink
Update python.js
Browse files Browse the repository at this point in the history
  • Loading branch information
lutzroeder committed Jun 6, 2024
1 parent 7e9a58d commit b967bd8
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions source/python.js
Original file line number Diff line number Diff line change
Expand Up @@ -4765,6 +4765,7 @@ python.Execution = class {
throw new python.Error(`Unsupported range(${JSON.stringify(start)}, ${JSON.stringify(stop)}, ${JSON.stringify(step)})`);
});
this.registerFunction('torch._C._nn.gelu');
this.registerFunction('torch._C._nn.avg_pool2d');
this.registerFunction('torch._native_multi_head_attention');
this.registerFunction('torch._utils._rebuild_sparse_tensor', (layout, data) => {
if (layout === torch.sparse_coo) {
Expand Down

0 comments on commit b967bd8

Please sign in to comment.