onnxruntime
f2dc725b - Add SpaceToDepth and DepthToSpace CUDA NHWC Ops (#19646)

Commit
1 year ago
Add SpaceToDepth and DepthToSpace CUDA NHWC Ops (#19646) ### Description - Adding CUDA NHWC support for SpaceToDepth and DepthToSpace - Add a new test which verifies that swizzling SpaceToDepth swizzling for the H axis is correct. - If CUDA NHWC is enabled, run all tests on the CUDA EP with NHWC as well. ### Motivation and Context Adding more NHWC operations to avoid layout transformations when using the CUDA EP for more efficiency.
Author
Parents
Loading