Searched refs:outDimIndex (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 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
[all …]
|