onnxruntime/js/web/lib/wasm/jsep
Jiajia Qin 64dacc2892
[js/webgpu] Add BatchNormalization Op (#18468)
### Description
This PR adds `BatchNormalization` with `float` support.

Some Todos:
1. all inputs don't have same data type. For example, x/y is float16,
but bias/scale is float32 or double.
2. training mode support.

We see many models are using `BatchNormalization` ops. However, due to
the missing in jsep, all of them run on cpu, which result very poor
performance. With this PR's support, densenet-9 model becomes 20.29 ms
from 250.69 ms.
2023-11-22 15:58:06 -08:00
..
webgpu [js/webgpu] Add BatchNormalization Op (#18468) 2023-11-22 15:58:06 -08:00
backend-webgpu.ts [js/webgpu] Use builtin num_workgroups to fix shader key conflict (#18387) 2023-11-10 17:37:45 -08:00
init.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
log.ts
tensor-view.ts [js/web] revise TensorView (#17473) 2023-09-14 21:14:44 -07:00
util.ts [js/webgpu] allow 0 sized tensor for tensor view (#16540) 2023-06-30 12:05:04 -07:00