onnxruntime/js/web/lib/wasm/jsep/webgpu/ops
Jiajia Qin 2e0a388c36
[js/webgpu] Add HardSigmoid support (#19215)
### Description
This op is required in mobilenetv3-small-100. With this PR,
mobilenetv3-small-100 model becomes less than 10 ms from over 100 ms on
ADL.
2024-01-22 15:53:26 -08:00
..
3rd-party
argminmax.ts
attention.ts [js/webgpu] Support uniforms for attention and multihead attention (#18903) 2024-01-09 07:46:30 -08:00
batch-norm.ts [js/webgpu] Change A/sqrt(B) to A*inverseSqrt(B) in normalization ops (#19101) 2024-01-12 00:08:16 -08:00
bias-add.ts
bias-split-gelu.ts
binary-op.ts
common.ts [js/webgpu] Support uniforms for attention and multihead attention (#18903) 2024-01-09 07:46:30 -08:00
concat.ts
conv-grouped.ts [js/webgpu] Provide a vectorized algorithm for GroupedConv (#18884) 2024-01-10 16:12:43 -08:00
conv-transpose.ts
conv.ts [js/webgpu] disable GroupedConvVectorize path (#19090) 2024-01-11 08:13:14 -08:00
cumsum.ts
einsum.ts
expand.ts [js/webgpu] Fix Expand/Gather when input type is bool (#18999) 2024-01-05 08:16:15 -08:00
fuse-utils.ts
gather-elements.ts
gather.ts [js/webgpu] Fix Expand/Gather when input type is bool (#18999) 2024-01-05 08:16:15 -08:00
gemm.ts fix gemm beta for fp16 (#19153) 2024-01-15 18:40:38 -08:00
instance-norm.ts [js/webgpu] Change A/sqrt(B) to A*inverseSqrt(B) in normalization ops (#19101) 2024-01-12 00:08:16 -08:00
layer-norm.ts [js/webgpu] Change A/sqrt(B) to A*inverseSqrt(B) in normalization ops (#19101) 2024-01-12 00:08:16 -08:00
matmul.ts
multi-head-attentiion.ts [js/webgpu] Support uniforms for attention and multihead attention (#18903) 2024-01-09 07:46:30 -08:00
pad.ts [js/webgpu] Support pad uniforms (#19057) 2024-01-09 09:34:56 -08:00
pool.ts
range.ts [js/webgpu] Support range uniforms (#19055) 2024-01-09 09:33:57 -08:00
reduce-shared.ts
reduce.ts
resize.ts fix resize for fp16 (#19110) 2024-01-12 13:44:28 -08:00
skip-layer-norm.ts [js/webgpu] Change A/sqrt(B) to A*inverseSqrt(B) in normalization ops (#19101) 2024-01-12 00:08:16 -08:00
slice.ts
softmax.ts
split.ts
tile.ts
transpose.ts
unary-op.ts [js/webgpu] Add HardSigmoid support (#19215) 2024-01-22 15:53:26 -08:00
where.ts [js/webgpu] fix bcast in where (#19009) 2024-01-11 12:13:24 -08:00