From 41a19ae1b5d381b21b5ea6938cc7b77fb33be16d Mon Sep 17 00:00:00 2001 From: Yulong Wang <7679871+fs-eire@users.noreply.github.com> Date: Fri, 5 May 2023 08:29:19 -0700 Subject: [PATCH] [js/webgpu] fix Transpose with non-float tensor (#15819) ### Description fix Transpose with non-float tensor. only register float type for Transpose. --- onnxruntime/core/providers/js/operators/transpose.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/core/providers/js/operators/transpose.cc b/onnxruntime/core/providers/js/operators/transpose.cc index 6803e6e7a2..763bcafc05 100644 --- a/onnxruntime/core/providers/js/operators/transpose.cc +++ b/onnxruntime/core/providers/js/operators/transpose.cc @@ -12,7 +12,7 @@ ONNX_OPERATOR_VERSIONED_KERNEL_EX( 1, 12, kJsExecutionProvider, (*KernelDefBuilder::Create()) - .TypeConstraint("T", DataTypeImpl::AllFixedSizeTensorTypes()), + .TypeConstraint("T", DataTypeImpl::GetTensorType()), Transpose); ONNX_OPERATOR_KERNEL_EX( @@ -21,7 +21,7 @@ ONNX_OPERATOR_KERNEL_EX( 13, kJsExecutionProvider, (*KernelDefBuilder::Create()) - .TypeConstraint("T", DataTypeImpl::AllFixedSizeTensorTypes()), + .TypeConstraint("T", DataTypeImpl::GetTensorType()), Transpose); } // namespace js