Searched refs:splitDim (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/armnn/delegate/classic/src/ |
| H A D | Split.hpp | 63 const unsigned int splitDim = ComputeWrappedIndex(axis, inputTensorInfo.GetNumDimensions()); in VisitSplitOperator() local
95 if (splitterDimSizes[splitDim] % numSplits != 0) in VisitSplitOperator()
100 numSplits, splitterDimSizes[splitDim], nodeIndex); in VisitSplitOperator()
103 splitterDimSizes[splitDim] /= numSplits; in VisitSplitOperator()
113 splitDescriptor.SetViewOriginCoord(j, splitDim, splitterDimSizes[splitDim] * j); in VisitSplitOperator()
200 …const unsigned int splitDim = ComputeWrappedIndex(axisTensorData[0], inputTensorInfo.GetNumDimensi… in VisitSplitVOperator() local
270 if (splitSum != armnn::numeric_cast<int>(inputTensorInfo.GetShape()[splitDim])) in VisitSplitVOperator()
280 …sorData[inferredIndex] = armnn::numeric_cast<int>(inputTensorInfo.GetShape()[splitDim]) - splitSum; in VisitSplitVOperator()
300 if (dimIdx == splitDim) in VisitSplitVOperator()
307 splitDescriptor.SetViewOriginCoord(j, splitDim, accumSplit); in VisitSplitVOperator()
|
| /aosp_15_r20/external/armnn/src/armnnTfLiteParser/ |
| H A D | TfLiteParser.cpp | 3966 …const unsigned int splitDim = armnnUtils::GetUnsignedAxis(inputTensorInfo.GetNumDimensions(), axis… in ParseSplit() local
3986 if (splitterDimSizes[splitDim] % numSplits != 0) in ParseSplit()
3990 splitterDimSizes[splitDim] /= numSplits; in ParseSplit()
4000 splitDesc.SetViewOriginCoord(j, splitDim, splitterDimSizes[splitDim] * j); in ParseSplit()
4085 const unsigned int splitDim = ComputeWrappedIndex(axis, inputTensorInfo.GetNumDimensions()); in ParseSplitV() local
4129 if (splitSum != armnn::numeric_cast<int>(inputTensorInfo.GetShape()[splitDim])) in ParseSplitV()
4136 … splitsData[inferIdx] = armnn::numeric_cast<int>(inputTensorInfo.GetShape()[splitDim]) - splitSum; in ParseSplitV()
4158 if (dimIdx == splitDim) in ParseSplitV()
4165 splitDesc.SetViewOriginCoord(j, splitDim, accumSplit); in ParseSplitV()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3889 uint32_t scoreDim[] = {19, 3}, roiDim[] = {19, 12}, splitDim[] = {2}; in boxWithNmsLimitOpTest() local
3894 getOpType(ANEURALNETWORKS_TENSOR_INT32, 1, splitDim), getOpType(scalarOperandCode), in boxWithNmsLimitOpTest()
|