Home
last modified time | relevance | path

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

/test/xts/acts/ai/mindspore/mindsporectest/src/
H A Dohos_common.cpp432 void PackNCHWToNHWCFp32(const char *src, char *dst, int batch, int plane, int channel) { in PackNCHWToNHWCFp32() argument
435 for (int hw = 0; hw < plane; hw++) { in PackNCHWToNHWCFp32()
436 int nhwc_index = n * channel * plane + hw * channel + c; in PackNCHWToNHWCFp32()
437 int nchw_index = n * channel * plane + c * plane + hw; in PackNCHWToNHWCFp32()
H A Dohos_common.h53 void PackNCHWToNHWCFp32(const char *src, char *dst, int batch, int plane, int channel);

Completed in 2 milliseconds