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

[js/webgpu] enable fp16 for tile (#20791)

Browse source
This commit is contained in:
Guenther Schmuelling 2024-05-23 16:59:39 -07:00 committed by GitHub
parent d1af19db9d
commit 0cf7caaff2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
onnxruntime/core/providers/js/operators/tile.cc
View file

@ -13,9 +13,7 @@ ONNX_OPERATOR_VERSIONED_KERNEL_EX(
12,
kJsExecutionProvider,
KernelDefBuilder()
.TypeConstraint("T", {DataTypeImpl::GetTensorType<float>(),
DataTypeImpl::GetTensorType<int32_t>(),
DataTypeImpl::GetTensorType<uint32_t>()})
.TypeConstraint("T", JsepSupportedDataTypes())
.TypeConstraint("T1", DataTypeImpl::GetTensorType<int64_t>())
.InputMemoryType(OrtMemTypeCPU, 1),
Tile);
@ -26,9 +24,7 @@ ONNX_OPERATOR_KERNEL_EX(
13,
kJsExecutionProvider,
KernelDefBuilder()
.TypeConstraint("T", {DataTypeImpl::GetTensorType<float>(),
DataTypeImpl::GetTensorType<int32_t>(),
DataTypeImpl::GetTensorType<uint32_t>()})
.TypeConstraint("T", JsepSupportedDataTypes())
.TypeConstraint("T1", DataTypeImpl::GetTensorType<int64_t>())
.InputMemoryType(OrtMemTypeCPU, 1),
Tile);