Searched defs:splitDim (Results 1 – 3 of 3) sorted by relevance
63 const unsigned int splitDim = ComputeWrappedIndex(axis, inputTensorInfo.GetNumDimensions()); in VisitSplitOperator() local200 …const unsigned int splitDim = ComputeWrappedIndex(axisTensorData[0], inputTensorInfo.GetNumDimensi… in VisitSplitVOperator() local
3966 …const unsigned int splitDim = armnnUtils::GetUnsignedAxis(inputTensorInfo.GetNumDimensions(), axis… in ParseSplit() local4085 const unsigned int splitDim = ComputeWrappedIndex(axis, inputTensorInfo.GetNumDimensions()); in ParseSplitV() local
3889 uint32_t scoreDim[] = {19, 3}, roiDim[] = {19, 12}, splitDim[] = {2}; in boxWithNmsLimitOpTest() local