Searched refs:inDimIndex (Results 1 – 1 of 1) sorted by relevance
1360 int64_t inDimIndex = outDimToInDimMap[outDimIndex]; in getExpandedOutputDimSize() local1362 for (auto otherDimIndex : reassocation[inDimIndex]) { in getExpandedOutputDimSize()1373 int64_t inDimSize = inStaticShape[inDimIndex]; in getExpandedOutputDimSize()1376 ? inDesc.size(b, loc, inDimIndex) in getExpandedOutputDimSize()1395 for (auto inDimIndex : reassocation[outDimIndex]) { in getCollapsedOutputDimSize() local1396 int64_t inDimSize = inStaticShape[inDimIndex]; in getCollapsedOutputDimSize()1399 ? inDesc.size(b, loc, inDimIndex) in getCollapsedOutputDimSize()