Searched refs:last_dim_index (Results 1 – 1 of 1) sorted by relevance
4745 int64_t last_dim_index = input_rank - 1; in matchAndRewrite() local4746 int64_t last_dim_size = input_type.getDimSize(last_dim_index); in matchAndRewrite()4753 op.getLoc(), iota_type, rewriter.getI64IntegerAttr(last_dim_index)); in matchAndRewrite()4758 op.getLoc(), llvm::ArrayRef<Value>{op.input(), iota_op}, last_dim_index, in matchAndRewrite()