From 0cf7caaff2ef95b7fbd2097d61981aec94e8ae9e Mon Sep 17 00:00:00 2001 From: Guenther Schmuelling Date: Thu, 23 May 2024 16:59:39 -0700 Subject: [PATCH] [js/webgpu] enable fp16 for tile (#20791) --- onnxruntime/core/providers/js/operators/tile.cc | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/onnxruntime/core/providers/js/operators/tile.cc b/onnxruntime/core/providers/js/operators/tile.cc index f27b6bae0c607..7fac0a4279d24 100644 --- a/onnxruntime/core/providers/js/operators/tile.cc +++ b/onnxruntime/core/providers/js/operators/tile.cc @@ -13,9 +13,7 @@ ONNX_OPERATOR_VERSIONED_KERNEL_EX( 12, kJsExecutionProvider, KernelDefBuilder() - .TypeConstraint("T", {DataTypeImpl::GetTensorType(), - DataTypeImpl::GetTensorType(), - DataTypeImpl::GetTensorType()}) + .TypeConstraint("T", JsepSupportedDataTypes()) .TypeConstraint("T1", DataTypeImpl::GetTensorType()) .InputMemoryType(OrtMemTypeCPU, 1), Tile); @@ -26,9 +24,7 @@ ONNX_OPERATOR_KERNEL_EX( 13, kJsExecutionProvider, KernelDefBuilder() - .TypeConstraint("T", {DataTypeImpl::GetTensorType(), - DataTypeImpl::GetTensorType(), - DataTypeImpl::GetTensorType()}) + .TypeConstraint("T", JsepSupportedDataTypes()) .TypeConstraint("T1", DataTypeImpl::GetTensorType()) .InputMemoryType(OrtMemTypeCPU, 1), Tile);