saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add missing file for pr20791 (#20811)

Browse source 
this file should have been in pr20791 to allow fp16 in the tile
implementation
This commit is contained in:
Guenther Schmuelling 2024-05-24 09:59:13 -07:00 committed by GitHub
parent 10c425a4d5
commit 33a68d221f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
js/web/lib/wasm/jsep/webgpu/ops/tile.ts
View file

@ -17,9 +17,9 @@ const validateInputs = (inputs: readonly TensorView[]): void => {
throw new Error('Tile requires 2 inputs.');
}
if (inputs[0].dataType !== DataType.float && inputs[0].dataType !== DataType.int32 &&
inputs[0].dataType !== DataType.uint32) {
throw new Error('Tile only support float, int32, and uint32 data types');
if (inputs[0].dataType !== DataType.float && inputs[0].dataType !== DataType.float16 &&
inputs[0].dataType !== DataType.int32 && inputs[0].dataType !== DataType.uint32) {
throw new Error('Tile only support float, float16, int32, and uint32 data types');
}
if (inputs[1].dataType !== DataType.int64) {