saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-16 21:00:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/js/web/test/data/ops
History
Satya Kumar Jandhyala fd8ee4894d
[JS/WebGPU] GroupQueryAttention rewrite (#20946) 
### Description
Implement JSEP GroupQueryAttention



### Motivation and Context
Required to enable certain LLM models to run using WebGPU.
2024-10-23 10:14:09 -07:00
..
_example.jsonc
abs-int32.jsonc [js/webgpu] Include Support for neg.int32 (#17374) 2023-09-06 12:00:16 -07:00
abs.jsonc
absr.jsonc
abss.jsonc
acos.jsonc
add.jsonc [js/web] fix suite test list for zero sized tensor (#19638) 2024-02-24 10:09:07 -08:00
add_int32.jsonc
add_zero-sized.jsonc [js/web] fix suite test list for zero sized tensor (#19638) 2024-02-24 10:09:07 -08:00
and.jsonc
asin.jsonc
attention.jsonc [js/web] JSEP Attention & MultiHeadAttention (#17742) 2023-11-17 12:23:52 -08:00
batch-norm.jsonc [js/webgpu] Add BatchNormalization Op (#18468) 2023-11-22 15:58:06 -08:00
bias-add.jsonc [js/web] BiasSplitGelu and BiasAdd kernels (#17161) 2023-10-03 12:20:20 -07:00
bias-split-gelu.jsonc [js/web] BiasSplitGelu and BiasAdd kernels (#17161) 2023-10-03 12:20:20 -07:00
cast.jsonc
ceil.jsonc
clip.jsonc [js/webgpu] support float16 for Clip (#21584) 2024-08-28 13:19:20 -07:00
concat.jsonc
concat_int32.jsonc [JS/WebGPU] support Concat.int32 operator (#17003) 2023-09-13 00:05:00 -07:00
concat_zero-sized.jsonc [JS/WebGPU] Preserve zero size input tensor dims. (#19737) 2024-03-07 19:07:49 -08:00
conv-transpose.jsonc [JS/WebGPU] Use non-matmul implementation for ConvTranspose in channel-first case. (#20022) 2024-03-23 11:19:14 -07:00
conv.jsonc [js/webgpu] Optimize grouped conv (#21892) 2024-09-04 17:16:35 -07:00
conv1d.jsonc [js/webgpu] Optimize conv1d by conv2d (#19388) 2024-08-22 22:56:07 -07:00
conv3dncdhw.jsonc [js/webgpu] Support conv3d naive (#20706) 2024-06-19 10:13:50 -07:00
cos.jsonc
cumsum.jsonc [js/webgpu] Fix shader compilation errors in cumsum (#18779) 2023-12-11 18:11:38 -08:00
depth-to-space.jsonc
dequantize-linear-int4.jsonc [JS/WebGPU] Add GatherBlockQuantized op support (#21734) 2024-08-26 14:46:04 -07:00
dequantizelinear.jsonc [JS/WebGPU] Add Dequantizelinear operator (#21642) 2024-08-09 14:44:19 -07:00
div.jsonc
div_int32.jsonc
einsum.jsonc [JS/Web] Add uniforms to Einsum (#18531) 2023-11-29 15:30:33 -08:00
equal.jsonc
exp.jsonc
expand.jsonc [js/webgpu] Fix Expand/Gather when input type is bool (#18999) 2024-01-05 08:16:15 -08:00
fast-gelu.jsonc [js/webgpu] support customop FastGelu (#19392) 2024-02-06 09:07:31 -08:00
floor.jsonc
fused-conv.jsonc [js/webgpu] Add activation Tanh (#21540) 2024-07-29 11:05:34 -07:00
fused-conv3dncdhw.jsonc [js/webgpu] Add activation for conv3d naive (#21466) 2024-07-29 08:47:41 -07:00
gather-block-quantized.jsonc [JS/WebGPU] Add GatherBlockQuantized op support (#21734) 2024-08-26 14:46:04 -07:00
gather-elements.jsonc
gather.jsonc [js/webgpu] Fix Expand/Gather when input type is bool (#18999) 2024-01-05 08:16:15 -08:00
gelu.jsonc
gemm.jsonc
global-average-pool.jsonc [js/webgpu] Support uniforms for pool (#18656) 2023-12-05 07:54:30 -08:00
greater.jsonc
group-query-attention.jsonc [JS/WebGPU] GroupQueryAttention rewrite (#20946) 2024-10-23 10:14:09 -07:00
identity.jsonc
image-scaler.jsonc
instance-norm.jsonc [js/webgpu] Fix NAN caused by un-initialized buffer in instance-norm (#19387) 2024-03-18 22:59:32 -07:00
layer-norm.jsonc
leaky-relu.jsonc
less.jsonc
log.jsonc
matmul-broadcast.jsonc [js/webgpu] add matmul broadcast tests (#17335) 2023-09-05 20:41:46 -07:00
matmul.jsonc
matmulnbits.jsonc [JS/WebGPU] Improve MatMulNBits perf (#19974) 2024-04-12 11:03:05 -07:00
max-pool.jsonc [js/webgpu] Fix max pool shape end with 0 (#21698) 2024-08-13 20:59:24 -07:00
mul.jsonc
mul_int32.jsonc
multihead-attention.jsonc [JS/WebGPU] Avoid producing presentKey/presentValue outputs if pastKey/pastValue … (#21782) 2024-08-19 18:02:19 -07:00
neg-int32.jsonc [js/webgpu] Include Support for neg.int32 (#17374) 2023-09-06 12:00:16 -07:00
neg.jsonc
not.jsonc
or.jsonc
pad-big.jsonc
pad.jsonc
pad_f16.jsonc [js/webgpu] Enable pad f16 uniform (#21691) 2024-08-26 07:58:48 -07:00
pow-big-number.jsonc
pow.jsonc
pow_int32.jsonc
quick-gelu.jsonc webgpu quickgelu (#20939) 2024-06-06 08:21:33 -07:00
reduce-min.jsonc
relu.jsonc
reshape-int32.jsonc
reshape-pack.jsonc
reshape.jsonc
resize-pack.jsonc
resize.jsonc [JS/Web] Sajandhy/webgpu resize scales rank check (#18954) 2023-12-29 09:23:27 -08:00
rotary-embedding.jsonc [js/webgpu] Implement com.microsoft.RotaryEmbedding (#20209) 2024-04-08 09:11:26 -07:00
shape.jsonc
simplified-layer-norm.jsonc [js/web] fix test runner with optional input/output (#20399) 2024-04-22 12:53:10 -07:00
sin.jsonc
skip-layer-norm.jsonc [js/web] fix test runner with optional input/output (#20399) 2024-04-22 12:53:10 -07:00
skip-simplified-layer-norm.jsonc [js/web] fix test runner with optional input/output (#20399) 2024-04-22 12:53:10 -07:00
slice.jsonc [JS/Web]Added uniforms support to Slice op. (#18422) 2023-11-16 09:44:13 -08:00
softmax.jsonc [js/webgpu] Remove the limitation on axis in softmax (#22231) 2024-09-30 18:27:11 -07:00
split.jsonc [js/webgpu] Handle negative axis in op Split (#21771) 2024-08-17 16:41:23 -07:00
sqrt.jsonc
sub.jsonc
sub_int32.jsonc
tan.jsonc
tanh.jsonc [js/webgpu] Fix Tanh explosion (#19201) 2024-01-25 08:25:35 -08:00
tile.jsonc
transpose.jsonc [js/webgpu] Optimize transpose (#21964) 2024-09-04 12:04:04 -07:00
transpose_int32_uint32.jsonc
upsample.jsonc [JS/WebGPU] Add trilinear interpolation to Resize; activation_params attribute is optional for FusedConv also. (#18842) 2023-12-27 16:21:29 -08:00
where.jsonc [JS/WebGPU] Fix Split and Where to handle corner cases. (#19613) 2024-02-23 00:21:15 -08:00
where_broadcast.jsonc [js/webgpu] Support where (#17544) 2023-10-03 14:28:21 -07:00
xor.jsonc