Home
last modified time | relevance | path

Searched defs: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
132 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
H A DConcatenation.cpp171 const uint32_t inputRank = getNumberOfDimensions(context->getInputShape(i)); in validate() local
H A DUnidirectionalSequenceLSTM.cpp201 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
H A DLSTM.cpp429 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat32() local
549 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat16() local
H A DSoftmax.cpp254 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate() local
H A DQLSTM.cpp180 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
H A DConv2D.cpp539 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate() local
/aosp12/packages/modules/NeuralNetworks/common/
H A DLegacyUtils.cpp794 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
1325 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
1375 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
1440 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
H A DValidation.cpp1842 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperationButNotOperandsImpl() local
2329 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperationButNotOperandsImpl() local
2375 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperationButNotOperandsImpl() local
2432 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperationButNotOperandsImpl() local