onnxruntime/onnxruntime/contrib_ops/js
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
..
bert [JS/WebGPU] GroupQueryAttention rewrite (#20946) 2024-10-23 10:14:09 -07:00
quantization [JS/WebGPU] Add GatherBlockQuantized op support (#21734) 2024-08-26 14:46:04 -07:00
bias_add.cc
bias_add.h
bias_split_gelu.cc
bias_split_gelu.h
fast_gelu.cc
fast_gelu.h
fused_conv.cc
gelu.cc
gelu.h
js_contrib_kernels.cc [JS/WebGPU] Add GatherBlockQuantized op support (#21734) 2024-08-26 14:46:04 -07:00
js_contrib_kernels.h
layer_norm.cc support for layernorm in webgpu pre opset-17 (#21121) 2024-06-27 10:20:48 -07:00
quick_gelu.cc webgpu quickgelu (#20939) 2024-06-06 08:21:33 -07:00
quick_gelu.h webgpu quickgelu (#20939) 2024-06-06 08:21:33 -07:00
skip_layer_norm.cc [js/web] support SimplifiedLayerNorm and SkipSimplifiedLayerNorm (#20277) 2024-04-11 14:08:50 -07:00
skip_layer_norm.h [js/web] support SimplifiedLayerNorm and SkipSimplifiedLayerNorm (#20277) 2024-04-11 14:08:50 -07:00