Searched refs:dim_exprs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 463 SmallVector<AffineExpr, 4> dim_exprs; in getIndexingMaps() local
464 dim_exprs.reserve(nloops); in getIndexingMaps()
472 dim_exprs.push_back(expansion_needed ? b->getAffineConstantExpr(0) in getIndexingMaps()
477 AffineMap::get(nloops, /*symbolCount=*/0, dim_exprs, b->getContext()), in getIndexingMaps()
670 SmallVector<AffineExpr, 4> dim_exprs; in getIndexingMaps() local
671 dim_exprs.reserve(nloops); in getIndexingMaps()
682 dim_exprs.push_back(b->getAffineDimExpr(size)); in getIndexingMaps()
685 AffineMap::get(nloops, /*symbolCount=*/0, dim_exprs, b->getContext()), in getIndexingMaps()
|