@@ -200,17 +200,17 @@ Status DepthToSpace::Compute(OpKernelContext* context) const {
200200 onnxruntime::narrow<std::ptrdiff_t >(blocksize_));
201201 } else if (input.IsDataType <uint8_t >()) {
202202 SpaceDepthOpCpuImpl<uint8_t >(input, output, permutation,
203- onnxruntime::narrow<std::ptrdiff_t >(batch),
204- onnxruntime::narrow<std::ptrdiff_t >(dim1),
205- onnxruntime::narrow<std::ptrdiff_t >(blocksize_),
206- onnxruntime::narrow<std::ptrdiff_t >(dim3),
207- onnxruntime::narrow<std::ptrdiff_t >(input_height),
208- onnxruntime::narrow<std::ptrdiff_t >(input_width),
209- onnxruntime::narrow<std::ptrdiff_t >(input_depth / blocksize_ / blocksize_),
210- onnxruntime::narrow<std::ptrdiff_t >(input_height),
211- onnxruntime::narrow<std::ptrdiff_t >(blocksize_),
212- onnxruntime::narrow<std::ptrdiff_t >(input_width),
213- onnxruntime::narrow<std::ptrdiff_t >(blocksize_));
203+ onnxruntime::narrow<std::ptrdiff_t >(batch),
204+ onnxruntime::narrow<std::ptrdiff_t >(dim1),
205+ onnxruntime::narrow<std::ptrdiff_t >(blocksize_),
206+ onnxruntime::narrow<std::ptrdiff_t >(dim3),
207+ onnxruntime::narrow<std::ptrdiff_t >(input_height),
208+ onnxruntime::narrow<std::ptrdiff_t >(input_width),
209+ onnxruntime::narrow<std::ptrdiff_t >(input_depth / blocksize_ / blocksize_),
210+ onnxruntime::narrow<std::ptrdiff_t >(input_height),
211+ onnxruntime::narrow<std::ptrdiff_t >(blocksize_),
212+ onnxruntime::narrow<std::ptrdiff_t >(input_width),
213+ onnxruntime::narrow<std::ptrdiff_t >(blocksize_));
214214 } else {
215215 // user will not see this as the kernel doesn't claim support for types other than float and double
216216 return ORT_MAKE_STATUS (ONNXRUNTIME, FAIL, " Unsupported input type in DepthToSpace op: " , input.DataType ());
0 commit comments