Home
last modified time | relevance | path

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

/aosp_15_r20/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
DPack.cpp63 if (const size_t inputRank = inputShape.dimensions.size()) { in validate() local
DConcatenation.cpp54 const uint32_t inputRank = getNumberOfDimensions(context->getInputShape(i)); in validate() local
DConv2D.cpp31 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate() local
/aosp_15_r20/external/armnn/delegate/classic/src/
H A DControl.hpp30 const uint32_t inputRank = concatDescriptor.GetNumDimensions(); in SetupConcatViewOrigin() local
99 uint32_t inputRank = tfLiteTensors[tfLiteNode->inputs->data[0]].dims->size; in VisitConcatenationOperator() local
H A DSlice.hpp43 int inputRank = tfLiteInputs[0]->dims->size; in VisitSliceOperator() local
H A DStridedSlice.hpp43 int inputRank = tfLiteInputs[0]->dims->size; in VisitStridedSliceOperator() local
/aosp_15_r20/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
DLSTM.cpp429 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat32() local
549 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat16() local
DUnidirectionalSequenceLSTM.cpp91 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
DMirrorPad.cpp42 const auto inputRank = getNumberOfDimensions(inputShape); in prepare() local
DQLSTM.cpp64 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
/aosp_15_r20/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
/aosp_15_r20/external/armnn/src/armnnUtils/
H A DParserHelper.cpp25 const uint32_t inputRank = concatDescriptor.GetNumDimensions(); in ProcessConcatInputTensorInfo() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Drestrict_max_rank.cc97 int64_t inputRank = inputTy.getRank(); in matchAndRewrite() local
H A Dlegalize_to_linalg.cc649 unsigned inputRank = inputType.getRank(); in getIndexingMaps() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc2750 int64_t inputRank = operandRankedType.getRank(); in inferReturnTypeComponents() local
4165 int64_t inputRank = inputTensor.getRank(); in matchAndRewrite() local
4244 int inputRank = inputType.getRank(); in verify() local
4291 int64_t inputRank = inputTy.getRank(); in matchAndRewrite() local
6004 int inputRank = inputType.getRank(); in verify() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp2233 int64_t inputRank = operandRankedType.getRank(); in inferReturnTypeComponents() local
3189 int inputRank = inputType.getRank(); in verify() local
4431 int inputRank = inputType.getRank(); in verify() local
/aosp_15_r20/external/armnn/src/armnnOnnxParser/
H A DOnnxParser.cpp1647 uint32_t inputRank = m_TensorsInfo[node.input(0)].m_info->GetNumDimensions(); in ParseConcat() local
/aosp_15_r20/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
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/
H A DTfLiteParser.cpp3234 uint32_t inputRank = InputTensorInfo(subgraphIndex, operatorIndex, 0).GetNumDimensions(); in ParseConcatenation() local