onnxruntime/js/web/lib/wasm/jsep/webgpu/ops
Arthur Islamov ea55700e1c
[js/web] JSEP Gather OP (#16855)
### Description
Added Gather op that works with both i32 and i64 indices, assuming that
values fall into i32 limit. The assumption is safe because it's not
possible to allocate more than 2gb buffer for inputs.

It treats all data from input tensor as u32, copying 1 or 2 elements for
i64, u64 and double.

---------

Co-authored-by: Guenther Schmuelling <guschmue@microsoft.com>
2023-08-03 14: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/webgpu: argmax,argmin,softmax support (#16882) 2023-08-02 18:16:19 -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] JSEP Expand fix for inputs with rank < 2 (#16829) 2023-08-03 11:38:04 -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
matmul.ts
pool.ts
reduce.ts [JS/Web] Fixed the output indexing in the shader code when the output is 1-dim. (#16508) 2023-06-30 09:42:38 -07:00
resize.ts [JS/WebGPU] Add Resize operator (#16680) 2023-07-31 09:35:06 -07:00
slice.ts [JS/WebGPU] Add Resize operator (#16680) 2023-07-31 09:35:06 -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