Searched defs:getIndexingMaps (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 412 static SmallVector<AffineMap, 2> getIndexingMaps(OpTy broadcast_op, in getIndexingMaps() function in mlir::__anon57cf929c0111::BroadcastConverter
449 static SmallVector<AffineMap, 2> getIndexingMaps( in getIndexingMaps() function in mlir::__anon57cf929c0111::HloBroadcastInDimConverter
656 SmallVector<AffineMap, 2> getIndexingMaps(lmhlo::BroadcastInDimOp op, in getIndexingMaps() function in mlir::__anon57cf929c0111::LhloBroadcastInDimConverter
697 static SmallVector<AffineMap, 2> getIndexingMaps(OpTy op, Builder* b) { in getIndexingMaps() function in mlir::__anon57cf929c0111::TransposeConverter
1018 static SmallVector<AffineMap, 2> getIndexingMaps(OpTy op, Builder* b) { in getIndexingMaps() function in mlir::__anon57cf929c0111::ReverseConverter
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorOps.cpp | 522 SmallVector<AffineMap, 4> ContractionOp::getIndexingMaps() { in getIndexingMaps() function in ContractionOp
|