Searched refs:concatenationDimension (Results 1 – 2 of 2) sorted by relevance
290 unsigned int concatenationDimension) in CreateDescriptorForConcatenation() argument310 if (concatenationDimension >= numDimensions) in CreateDescriptorForConcatenation()319 … const bool dimSizeOk = (d == concatenationDimension) || (firstInputShape[d] == (*it)[d]); in CreateDescriptorForConcatenation()329 viewsDescriptor.SetConcatAxis(concatenationDimension); in CreateDescriptorForConcatenation()337 for (unsigned int i = 0; i < concatenationDimension; ++i) in CreateDescriptorForConcatenation()342 viewsDescriptor.SetViewOriginCoord(viewIndex, concatenationDimension, coordAlongConcatDim); in CreateDescriptorForConcatenation()343 unsigned int dimSize = inputShape[concatenationDimension]; in CreateDescriptorForConcatenation()347 for (unsigned int i = concatenationDimension + 1; i < numDimensions; ++i) in CreateDescriptorForConcatenation()
1243 unsigned int concatenationDimension) in CreateDescriptorForConcatenation() argument1245 …urn armnn::CreateDescriptorForConcatenation(shapes.begin(), shapes.end(), concatenationDimension); in CreateDescriptorForConcatenation()1260 unsigned int concatenationDimension);