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

microsoft · May 23, 2024 · 0cf7caa · 0cf7caa
1 parent d1af19d
commit 0cf7caa
Showing 1 changed file with 2 additions and 6 deletions.
diff --git 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<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);