Lines Matching refs:outDimIndex
1349 int64_t outDimIndex, ArrayRef<int64_t> outStaticShape, in getExpandedOutputDimSize() argument
1354 int64_t outDimSize = outStaticShape[outDimIndex]; in getExpandedOutputDimSize()
1360 int64_t inDimIndex = outDimToInDimMap[outDimIndex]; in getExpandedOutputDimSize()
1363 if (otherDimIndex == static_cast<unsigned>(outDimIndex)) in getExpandedOutputDimSize()
1387 OpBuilder &b, Location loc, Type &llvmIndexType, int64_t outDimIndex, in getCollapsedOutputDimSize() argument
1395 for (auto inDimIndex : reassocation[outDimIndex]) { in getCollapsedOutputDimSize()
1415 llvm::seq<int64_t>(0, outStaticShape.size()), [&](int64_t outDimIndex) { in getCollapsedOutputShape() argument
1416 return getCollapsedOutputDimSize(b, loc, llvmIndexType, outDimIndex, in getCollapsedOutputShape()
1417 outStaticShape[outDimIndex], in getCollapsedOutputShape()
1431 llvm::seq<int64_t>(0, outStaticShape.size()), [&](int64_t outDimIndex) { in getExpandedOutputShape() argument
1432 return getExpandedOutputDimSize(b, loc, llvmIndexType, outDimIndex, in getExpandedOutputShape()