Add uint8_t support for DepthToSpace 11

microsoft · Dec 12, 2024 · cd6ec1b · cd6ec1b
1 parent e00b426
commit cd6ec1b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/onnxruntime/core/providers/cpu/tensor/space_depth_ops.cc b/onnxruntime/core/providers/cpu/tensor/space_depth_ops.cc
@@ -43,7 +43,8 @@ ONNX_CPU_OPERATOR_VERSIONED_KERNEL(
     12,
     KernelDefBuilder()
         .TypeConstraint("T", {DataTypeImpl::GetTensorType<float>(),
-                              DataTypeImpl::GetTensorType<double>()}),
+                              DataTypeImpl::GetTensorType<double>(),
+                              DataTypeImpl::GetTensorType<uint8_t>()}),
     DepthToSpace);
 
 ONNX_CPU_OPERATOR_KERNEL(