[Feature Request] SpaceToDepth & DepthToSpace integer implementations #21287
Labels
contributions welcome
lower priority issues for the core ORT teams
feature request
request for unsupported feature or enhancement
Describe the feature request
The
DepthToSpace
andSpaceToDepth
ops support integer types:https://github.com/onnx/onnx/blob/38afbd31ac9a585abb7463dcaaae121651e0a2d7/docs/Operators.md#DepthToSpace
https://github.com/onnx/onnx/blob/38afbd31ac9a585abb7463dcaaae121651e0a2d7/docs/Operators.md#SpaceToDepth
There's currently only implementations for
float
anddouble
in CPU:https://github.com/microsoft/onnxruntime/blob/4c3c809bdbcde4ea96f0a31a242ca6877a10c40a/onnxruntime/core/providers/cpu/tensor/space_depth_ops.cc
and float16 in CUDA:
https://github.com/microsoft/onnxruntime/blob/4c3c809bdbcde4ea96f0a31a242ca6877a10c40a/onnxruntime/core/providers/cuda/tensor/space_depth_ops.cc
Describe scenario use case
Running quantized models
The text was updated successfully, but these errors were encountered: