Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DReduce.cpp56 const uint32_t inputRank = getNumberOfDimensions(inputShape); in compute() local
62 reinterpret_cast<const int32_t*>(inputShape.dimensions.data()), inputRank, in compute()
132 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
133 NN_RET_CHECK_LE(inputRank, 4); in prepare()
135 std::vector<bool> shouldReduce(inputRank); in prepare()
142 NN_RET_CHECK(handleNegativeAxis(inputRank, &axis)); in prepare()
150 for (uint32_t axis = 0; axis < inputRank; ++axis) { in prepare()
H A DSoftmax.cpp254 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate() local
255 if (inputRank != 0) { in validate()
256 NN_RET_CHECK_LE(inputRank, 4); in validate()
262 if (inputRank != 2 && inputRank != 4 && inputRank != 0) { in validate()
H A DLSTM.cpp429 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat32() local
430 NN_CHECK(inputRank == 2 || inputRank == 3); in LSTMEvalFloat32()
433 (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 0 : 1) : 1; in LSTMEvalFloat32()
434 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat32()
436 const uint32_t inputSize = getSizeOfDimension(input_shape, inputRank - 1); in LSTMEvalFloat32()
549 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat16() local
550 NN_CHECK(inputRank == 2 || inputRank == 3); in LSTMEvalFloat16()
553 (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 0 : 1) : 1; in LSTMEvalFloat16()
554 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat16()
556 const uint32_t inputSize = getSizeOfDimension(input_shape, inputRank - 1); in LSTMEvalFloat16()
H A DConcatenation.cpp171 const uint32_t inputRank = getNumberOfDimensions(context->getInputShape(i)); in validate() local
172 if (inputRank != 0) { in validate()
173 NN_RET_CHECK_LE(inputRank, 4); in validate()
H A DUnidirectionalSequenceLSTM.cpp201 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
202 NN_RET_CHECK_EQ(inputRank, 3) << "Invalid input tensor rank: " << inputRank; in prepare()
206 const uint32_t inputSize = getSizeOfDimension(inputShape, inputRank - 1); in prepare()
H A DQLSTM.cpp180 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
181 NN_RET_CHECK_EQ(inputRank, 2) << "Invalid input tensor rank: " << inputRank; in prepare()
H A DConv2D.cpp539 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate() local
541 if (inputRank != 0) { in validate()
542 NN_RET_CHECK_EQ(inputRank, 4); in validate()
/aosp12/packages/modules/NeuralNetworks/common/
H A DLegacyUtils.cpp794 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
795 if (inputRank > 4) { in validateOperation()
1325 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
1326 if (inputRank > 4) { in validateOperation()
1375 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
1376 if (inputRank > 4) { in validateOperation()
1440 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
1441 if (inputRank > 4) { in validateOperation()
H A DValidation.cpp1842 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperationButNotOperandsImpl() local
1843 NN_VALIDATE_LE(inputRank, 4u) in validateOperationButNotOperandsImpl()
2329 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperationButNotOperandsImpl() local
2330 NN_VALIDATE_LE(inputRank, 4u) in validateOperationButNotOperandsImpl()
2375 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperationButNotOperandsImpl() local
2376 NN_VALIDATE_LE(inputRank, 4u) in validateOperationButNotOperandsImpl()
2432 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperationButNotOperandsImpl() local
2433 NN_VALIDATE_LE(inputRank, 4u) in validateOperationButNotOperandsImpl()