Searched refs:dstIndices (Results 1 – 3 of 3) sorted by relevance
69 SmallVector<uint64_t, 6> dstIndices(outputType.getRank(), 0); in matchAndRewrite() local71 dstIndices[dim] = srcIndices[permValues[dim]]; in matchAndRewrite()73 uint64_t dstLinearIndex = dstIndices.front(); in matchAndRewrite()75 dstLinearIndex = dstLinearIndex * outputShape[dim] + dstIndices[dim]; in matchAndRewrite()
165 SmallVector<uint64_t> dstIndices(loopBounds.size(), 0); in matchAndRewrite() local193 dstIndices[dim] = indices[outputDims[dim]]; in matchAndRewrite()196 dstLinearIndex = dstIndices.front(); in matchAndRewrite()201 dstLinearIndex = dstLinearIndex * outputShape[dim] + dstIndices[dim]; in matchAndRewrite()
172 Variadic<Index>:$dstIndices,178 $src `[` $srcIndices `]` `,` $dst `[` $dstIndices `]` `,` $numElements