saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-20 02:07:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/js/web/lib/wasm/jsep/webgpu/ops
History
Arthur Islamov c3f04251c7
[js/web] JSEP LayerNormalization and InstanceNormalizations kernels (#16830) 
### Description
Added two kernels for Layer and Instance norm

Also added maximum limits for `maxBufferSize` when requesting GPU device
as by default it's limited to 256mb and it fails allocating 600mb buffer
while running fp32 StableDiffusion weights.


### Motivation and Context
These two are used in StableDiffusion and many other networks
2023-08-08 09:09:37 -07:00
..
3rd-party [JS/Web] Added Gelu contrib operator support to JSEP (#16909) 2023-07-31 09:18:58 -07:00
argminmax.ts [JS/Web] Fixed ArgMin and ArgMax and refactored (#17002) 2023-08-04 12:59:36 -07:00
binary-op.ts
common.ts
concat.ts [Web/JS] Support WebGPU Concat operator (#16543) 2023-07-05 11:59:45 -07:00
conv-grouped.ts
conv-transpose.ts [Web/JS] Add ConvTranspose support (#16433) 2023-07-08 11:10:50 -07:00
conv.ts [js/web] [JSEP] allow passing data in kernel compute (#16621) 2023-07-07 14:27:30 -07:00
conv2d-mm.ts
expand.ts [JS/Web] Modify Reduce, Expand and Slice to pass op and node tests. (#16979) 2023-08-03 15:48:47 -07:00
fuse-utils.ts
gather.ts [js/web] JSEP Gather OP (#16855) 2023-08-03 14:09:37 -07:00
gelu.ts [JS/Web] Added Gelu contrib operator support to JSEP (#16909) 2023-07-31 09:18:58 -07:00
gemm.ts
instance-norm.ts [js/web] JSEP LayerNormalization and InstanceNormalizations kernels (#16830) 2023-08-08 09:09:37 -07:00
layer-norm.ts [js/web] JSEP LayerNormalization and InstanceNormalizations kernels (#16830) 2023-08-08 09:09:37 -07:00
matmul.ts
pool.ts
reduce.ts [JS/Web] Fixed ArgMin and ArgMax and refactored (#17002) 2023-08-04 12:59:36 -07:00
resize.ts [JS/WebGPU] Add Resize operator (#16680) 2023-07-31 09:35:06 -07:00
slice.ts [JS/Web] Modify Reduce, Expand and Slice to pass op and node tests. (#16979) 2023-08-03 15:48:47 -07:00
softmax.ts js/webgpu: argmax,argmin,softmax support (#16882) 2023-08-02 18:16:19 -07:00
split.ts [Web/JS] Added Split operator support. (#16567) 2023-07-07 12:16:10 -07:00
transpose.ts [js/WebGPU] Support int32 Transpose in WebGPU (#16952) 2023-08-02 16:27:24 -07:00
unary-op.ts [JS/Web] Added Gelu contrib operator support to JSEP (#16909) 2023-07-31 09:18:58 -07:00