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

[js/webgpu] fix Transpose with non-float tensor (#15819)

Browse source 
### Description
fix Transpose with non-float tensor.

only register float type for Transpose.
This commit is contained in:
Yulong Wang 2023-05-05 08:29:19 -07:00 committed by GitHub
parent e139ae238b
commit 41a19ae1b5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/providers/js/operators/transpose.cc
View file

@ -12,7 +12,7 @@ ONNX_OPERATOR_VERSIONED_KERNEL_EX(
1, 12,
kJsExecutionProvider,
(*KernelDefBuilder::Create())
.TypeConstraint("T", DataTypeImpl::AllFixedSizeTensorTypes()),
.TypeConstraint("T", DataTypeImpl::GetTensorType<float>()),
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<float>()),
Transpose);
} // namespace js