Home
last modified time | relevance | path

Searched defs:inputRank (Results 1 – 8 of 8) sorted by relevance

/frameworks/ml/nn/common/operations/
DReduce.cpp56 const uint32_t inputRank = getNumberOfDimensions(inputShape); in compute() local
130 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
DConcatenation.cpp167 const uint32_t inputRank = getNumberOfDimensions(context->getInputShape(i)); in validate() local
DUnidirectionalSequenceLSTM.cpp198 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
DSoftmax.cpp250 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate() local
DLSTM.cpp429 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat32() local
549 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat16() local
DQLSTM.cpp179 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
DConv2D.cpp537 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate() local
/frameworks/ml/nn/common/
DUtils.cpp923 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
1469 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
1521 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
1587 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local