Searched refs:fusedTensorIndex (Results 1 – 1 of 1) sorted by relevance
411 unsigned fusedTensorIndex) { in isFusableWithReshapeByDimExpansion() argument426 linalgOp.getIndexingMap(fusedTensorIndex).getNumResults() > 0 && in isFusableWithReshapeByDimExpansion()438 unsigned fusedTensorIndex, in fuseWithReshapeByExpansion() argument440 assert(isFusableWithReshapeByDimExpansion(linalgOp, fusedTensorIndex) && in fuseWithReshapeByExpansion()447 AffineMap fusedIndexMap = linalgOp.getIndexingMap(fusedTensorIndex); in fuseWithReshapeByExpansion()547 if (operand.index() == fusedTensorIndex) { in fuseWithReshapeByExpansion()