saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-02 23:39:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/js/web/lib/wasm/jsep/webgpu/ops
History
Jiajia Qin cd3fb377ea
[js/webgpu] Allow binary ops with scalar to use the vectorize path (#17589) 
### Description
1. For binary ops, the components is always 4. So the dispatchGroup
should be : `{x: Math.ceil(outputSize / 64 /* workgroup size */ / 4 /*
component size */)}` instead of `{x: Math.ceil(outputSize / 64 /*
workgroup size */ / (vectorize ? 4 : 1) /* vec size */)}`.

2. If any of a or b only has one element, we still can use the vectorize
path since the same value will be broadcasted.
2023-09-21 20:55:08 -07:00
..
3rd-party [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
argminmax.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
binary-op.ts [js/webgpu] Allow binary ops with scalar to use the vectorize path (#17589) 2023-09-21 20:55:08 -07:00
common.ts [js/webgpu] Optimize InstanceNormalization (#17491) 2023-09-14 17:03:18 -07:00
concat.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
conv-grouped.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
conv-transpose.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
conv.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
conv2d-mm.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
einsum.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
expand.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
fuse-utils.ts
gather-elements.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
gather.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
gemm.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
instance-norm.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
layer-norm.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
matmul.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
pad.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
pool.ts [js/web] fp16 Pool & Reduce (#17512) 2023-09-21 14:52:13 -07:00
reduce.ts [js/web] fp16 Pool & Reduce (#17512) 2023-09-21 14:52:13 -07:00
resize.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
skip-layer-norm.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
slice.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
softmax.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
split.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
tile.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
transpose.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
unary-op.ts [js/web] FP16 binary and unary ops (#17515) 2023-09-18 15:43:32 -07:00