Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/types/operations/src/
DReshape.cpp148 const auto inputRank = context->getInputShape(0).dimensions.size(); in validatePad() local
195 const auto inputRank = context->getInputShape(0).dimensions.size(); in validatePadV2() local
340 const auto inputRank = context->getInputShape(0).dimensions.size(); in validateReshape() local
DSoftmax.cpp46 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate() local
DSimpleMath.cpp29 const auto inputRank = context->getInputShape(0).dimensions.size(); in validate() local
DConcatenation.cpp54 const uint32_t inputRank = getNumberOfDimensions(context->getInputShape(i)); in validate() local
DPack.cpp63 if (const size_t inputRank = inputShape.dimensions.size()) { in validate() local
DConv2D.cpp31 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate() local
/packages/modules/NeuralNetworks/common/cpu_operations/
DReduce.cpp49 const uint32_t inputRank = getNumberOfDimensions(inputShape); in compute() local
67 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
DUnidirectionalSequenceLSTM.cpp91 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
DMirrorPad.cpp42 const auto inputRank = getNumberOfDimensions(inputShape); in prepare() 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.cpp64 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
/packages/modules/NeuralNetworks/common/
DLegacyUtils.cpp766 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
1297 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
1347 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
1412 const auto inputRank = operands[inputIndexes[0]].dimensions.size(); in validateOperation() local
/packages/modules/NeuralNetworks/runtime/test/
DTestValidateOperations.cpp4725 constexpr size_t inputRank = sizeof(inputDimensions) / sizeof(inputDimensions[0]); in packTest() local
4778 constexpr size_t inputRank = sizeof(inputDimensions) / sizeof(inputDimensions[0]); in packTestBadQuantization() local
4839 constexpr size_t inputRank = sizeof(inputDimensions) / sizeof(inputDimensions[0]); in packTestBadRank() local