Home
last modified time | relevance | path

Searched refs:nchw_index (Results 1 - 1 of 1) sorted by relevance

/third_party/mindspore/test/utils/
H A Dcommon.cpp419 int nchw_index = n * channel * plane + c * plane + hw; in PackNCHWToNHWCFp32() local
420 dst[nhwc_index * 4] = src[nchw_index * 4]; in PackNCHWToNHWCFp32()
421 dst[nhwc_index * 4 + 1] = src[nchw_index * 4 + 1]; in PackNCHWToNHWCFp32()
422 dst[nhwc_index * 4 + 2] = src[nchw_index * 4 + 2]; in PackNCHWToNHWCFp32()
423 dst[nhwc_index * 4 + 3] = src[nchw_index * 4 + 3]; in PackNCHWToNHWCFp32()

Completed in 2 milliseconds