Jiajie Hu
447a3a7c70
[js/webgpu] Fix Expand/Gather when input type is bool ( #18999 )
...
### Description
Also update the op test suite.
### Motivation and Context
Previously the *total* size in case `Expand - last dim is not divisible
by 4` was a multiple of 4, even though the *last dimension* was not, so
the bug has never been caught.
2024-01-05 08:16:15 -08:00
Jiajia Qin
6781b6cf3d
[js/webgpu] add bool type for Expand/Gather ( #18615 )
...
### Description
In [detr-resnet-50](https://huggingface.co/Xenova/detr-resnet-50 ) model,
it uses expand with bool type running on cpu ep.
| Kernel | Shape | Provider |
| -------- | ------- | ------- |
| Expand | "input_type_shape" :
[{"bool":[1,1,1,625]},{"int64":[4]}],"activation_size" :
"657","output_type_shape" : [{"bool":[1,1,625,625]}] |
CPUExecutionProvider |
After this change, it will run on jsep.
| Kernel | Shape | Provider |
| -------- | ------- | ------- |
| Expand | "input_type_shape" :
[{"bool":[1,1,1,625]},{"int64":[4]}],"activation_size" :
"657","output_type_shape" : [{"bool":[1,1,625,625]}] |
JsExecutionProvider |
2023-11-30 15:47:08 -08:00
