Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/include/
DOperationsUtils.h346 const int secondDim = getSizeOfDimension(shape, 1); in transposeFirstTwoDimensions() local
353 for (int j = 0; j < secondDim; ++j) { in transposeFirstTwoDimensions()
356 buffer[(i * secondDim + j) * blockSize + k]; in transposeFirstTwoDimensions()
/frameworks/ml/nn/runtime/test/
DTestUnknownDimensions.cpp282 size_t secondDim = (second == DimensionKind::UNKNOWN_AT_COMPILE_OTHER_AT_EXECUTE) in TestOne() local
285 return firstDim * secondDim * sizeof(fives[0][0]); in TestOne()