Home
last modified time | relevance | path

Searched refs:permSize (Results 1 – 1 of 1) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DTranspose.cpp54 int32_t permSize = perm == nullptr ? 2 : static_cast<int32_t>(getSizeOfDimension(permShape, 0)); in transposeGeneric() local
60 for (int32_t output_k = 0, input_k = permSize - 1; output_k < permSize; ++output_k, --input_k) { in transposeGeneric()
61 reversed_perm[output_k] = permSize - perm[input_k] - 1; in transposeGeneric()
63 for (int32_t k = permSize; k < kOutputDimensionNum; ++k) { in transposeGeneric()