From 0e89ebc9b66423c565ead7b9f1318c39996dcdc2 Mon Sep 17 00:00:00 2001 From: Xu Xing Date: Tue, 16 Jul 2024 09:33:26 +0800 Subject: [PATCH] Revert "[js/webgpu] Enable graph capture for whisper-tiny-decoder" This reverts commit a10d717c0423c0b77e36fda75a6802c98b3f4f04. --- onnxruntime/core/providers/js/operators/reshape.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/onnxruntime/core/providers/js/operators/reshape.cc b/onnxruntime/core/providers/js/operators/reshape.cc index 7e8f6c651da5d..bd6772649a1ae 100644 --- a/onnxruntime/core/providers/js/operators/reshape.cc +++ b/onnxruntime/core/providers/js/operators/reshape.cc @@ -13,7 +13,7 @@ ONNX_OPERATOR_KERNEL_EX( 14, kJsExecutionProvider, (*KernelDefBuilder::Create()) - .TypeConstraint("T", DataTypeImpl::AllTensorTypes()) + .TypeConstraint("T", JsepSupportedDataTypes()) .TypeConstraint("shape", DataTypeImpl::GetTensorType()) .Alias(0, 0) .InputMemoryType(OrtMemTypeCPU, 1), @@ -25,7 +25,7 @@ ONNX_OPERATOR_VERSIONED_KERNEL_EX( 13, 13, kJsExecutionProvider, (*KernelDefBuilder::Create()) - .TypeConstraint("T", DataTypeImpl::AllTensorTypes()) + .TypeConstraint("T", JsepSupportedDataTypes()) .TypeConstraint("shape", DataTypeImpl::GetTensorType()) .Alias(0, 0) .InputMemoryType(OrtMemTypeCPU, 1), @@ -37,7 +37,7 @@ ONNX_OPERATOR_VERSIONED_KERNEL_EX( 5, 12, kJsExecutionProvider, (*KernelDefBuilder::Create()) - .TypeConstraint("T", DataTypeImpl::AllTensorTypes()) + .TypeConstraint("T", JsepSupportedDataTypes()) .TypeConstraint("shape", DataTypeImpl::GetTensorType()) .Alias(0, 0) .InputMemoryType(OrtMemTypeCPU, 1),