saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-05 04:17:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/js/web/lib/wasm/jsep/webgpu/ops
History
Yulong Wang 50e6235af1
[js/web] allow ShaderHelper to use internal (non-I/O) variables (#18525) 
### Description
This PR includes a change that inspired from #18452 to resolve a
requirement: a shader may depend on an instance of `IndicesHelper` to
generate WGSL code snippet, but the IndicesHelper instance is not
necessarily an input/output of the program. So the existing
`declareVariables()` function does not work with this scenario.

In order to support this requirement, I added this "use" function to
`interface ShaderHelper`, which takes a helper-like object as parameter.
The hidden implementation `ShaderHelperImpl` class will iterate the
helpers and call `impl()` for each.

@axinging @qjia7
2023-11-28 15:15:59 -08:00
..
3rd-party [js/web] allow ShaderHelper to use internal (non-I/O) variables (#18525) 2023-11-28 15:15:59 -08:00
argminmax.ts [js/web] set noUnusedParameters to true and fix a few bugs (#18404) 2023-11-15 09:16:29 -08:00
attention.ts [js/web] JSEP Attention & MultiHeadAttention (#17742) 2023-11-17 12:23:52 -08:00
batch-norm.ts [js/webgpu] Add BatchNormalization Op (#18468) 2023-11-22 15:58:06 -08:00
bias-add.ts [js/webgpu] revise uniform support (#17871) 2023-10-11 16:41:46 -07:00
bias-split-gelu.ts [JS/Web] Resize & BiasSplitGelu fp16 support (#18536) 2023-11-22 12:12:07 -08:00
binary-op.ts [js/webgpu] Optimize broadcast binary. (#18185) 2023-11-20 16:52:17 -08:00
common.ts [js/web] allow ShaderHelper to use internal (non-I/O) variables (#18525) 2023-11-28 15:15:59 -08:00
concat.ts [js/webgpu] Add uniforms support to concat op (#18238) 2023-11-10 13:46:03 -08:00
conv-grouped.ts [js/webgpu] Fix conv2d with activation (#18388) 2023-11-10 12:54:35 -08:00
conv-transpose.ts [js/webgpu] Fix the transpose error when dims > 4D (#18027) 2023-10-23 11:02:19 -07:00
conv.ts [js/webgpu] Fix the transpose error when dims > 4D (#18027) 2023-10-23 11:02:19 -07:00
einsum.ts [js/web] set noUnusedParameters to true and fix a few bugs (#18404) 2023-11-15 09:16:29 -08:00
expand.ts [[JS/Web]Added uniform to Expand op. (#18558) 2023-11-22 14:14:24 -08:00
fuse-utils.ts [js/webgpu] Fix conv2d with activation (#18388) 2023-11-10 12:54:35 -08:00
gather-elements.ts [js/webgpu] revise uniform support (#17871) 2023-10-11 16:41:46 -07:00
gather.ts [js/webgpu] Support uniforms for gather (#18312) 2023-11-13 11:24:34 -08:00
gemm.ts [js/webgpu] revise uniform support (#17871) 2023-10-11 16:41:46 -07:00
instance-norm.ts [js/web] FP16 LayerNorm, InstanceNorm, SkipLayerNorm (#17630) 2023-10-18 10:47:41 -07:00
layer-norm.ts [js/web] FP16 LayerNorm, InstanceNorm, SkipLayerNorm (#17630) 2023-10-18 10:47:41 -07:00
matmul.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
multi-head-attentiion.ts [js/web] JSEP Attention & MultiHeadAttention (#17742) 2023-11-17 12:23:52 -08:00
pad.ts [js/web] set noUnusedParameters to true and fix a few bugs (#18404) 2023-11-15 09:16:29 -08:00
pool.ts [JS/Web] Enabled 1d spacial input to GlobalAveragePool (#17973) 2023-10-23 16:02:50 -07:00
range.ts [js/webgpu] revise uniform support (#17871) 2023-10-11 16:41:46 -07:00
reduce-shared.ts [js/web] set noUnusedParameters to true and fix a few bugs (#18404) 2023-11-15 09:16:29 -08:00
reduce.ts [js/web] optimize reduce related operators (#17957) 2023-11-02 12:51:48 -07:00
resize.ts [JS/Web] Resize & BiasSplitGelu fp16 support (#18536) 2023-11-22 12:12:07 -08:00
skip-layer-norm.ts [js/web] FP16 LayerNorm, InstanceNorm, SkipLayerNorm (#17630) 2023-10-18 10:47:41 -07:00
slice.ts [JS/Web]Added uniforms support to Slice op. (#18422) 2023-11-16 09:44:13 -08:00
softmax.ts [js/webgpu] Support uniform for softmax (#18345) 2023-11-09 11:19:23 -08:00
split.ts [js/webgpu] revise uniform support (#17871) 2023-10-11 16:41:46 -07:00
tile.ts [js/webgpu] revise uniform support (#17871) 2023-10-11 16:41:46 -07:00
transpose.ts [js/webgpu] Fix the transpose error when dims > 4D (#18027) 2023-10-23 11:02:19 -07:00
unary-op.ts [js/web] unify resolve rules for "Clip" (#18527) 2023-11-20 23:18:06 -08:00
where.ts [js/webgpu] revise uniform support (#17871) 2023-10-11 16:41:46 -07:00