Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc742 unsigned curr_src_dim = 0, curr_dst_dim = 0; in matchAndRewrite() local
750 while (curr_src_dim < src_shape.size() && curr_dst_dim < dst_shape.size()) { in matchAndRewrite()
752 int64_t src_size = src_shape[curr_src_dim]; in matchAndRewrite()
753 while (src_size < dst_size && curr_src_dim < src_shape.size()) { in matchAndRewrite()
755 rewriter.getAffineDimExpr(curr_src_dim++)); in matchAndRewrite()
756 src_size *= src_shape[curr_src_dim]; in matchAndRewrite()
760 rewriter.getAffineDimExpr(curr_src_dim++)); in matchAndRewrite()
765 while (curr_src_dim < src_shape.size() && in matchAndRewrite()
766 src_shape[curr_src_dim] == 1) { in matchAndRewrite()
768 rewriter.getAffineDimExpr(curr_src_dim++)); in matchAndRewrite()
[all …]