Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/include/
DOperationsUtils.h329 const int firstDim = getSizeOfDimension(shape, 0); in transposeFirstTwoDimensions() local
336 for (int i = 0; i < firstDim; ++i) { in transposeFirstTwoDimensions()
339 transposedBuffer[(j * firstDim + i) * blockSize + k] = in transposeFirstTwoDimensions()
/packages/modules/NeuralNetworks/runtime/test/
DTestUnknownDimensions.cpp284 size_t firstDim = (first == DimensionKind::UNKNOWN_AT_COMPILE_OTHER_AT_EXECUTE) in TestOne() local
290 return firstDim * secondDim * sizeof(fives[0][0]); in TestOne()