Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DTranspose.cpp49 int32_t permSize = perm == nullptr ? 2 : static_cast<int32_t>(getSizeOfDimension(permShape, 0)); in transposeGeneric() local
55 for (int32_t output_k = 0, input_k = permSize - 1; output_k < permSize; ++output_k, --input_k) { in transposeGeneric()
56 reversed_perm[output_k] = permSize - perm[input_k] - 1; in transposeGeneric()
58 for (int32_t k = permSize; k < kOutputDimensionNum; ++k) { in transposeGeneric()