Home
last modified time | relevance | path

Searched refs:sourceDimPos (Results 1 – 1 of 1) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorInferTypeOpInterfaceImpl.cpp81 unsigned sourceDimPos = expandedDimToCollapsedDim[dimIndex]; in getExpandedOutputDimFromInputShape() local
82 unsigned startPos = reassociation[sourceDimPos] in getExpandedOutputDimFromInputShape()
87 unsigned endPos = reassociation[sourceDimPos] in getExpandedOutputDimFromInputShape()
102 Value sourceDim = builder.create<tensor::DimOp>(loc, src, sourceDimPos); in getExpandedOutputDimFromInputShape()