Searched defs:broadcast_op (Results 1 – 2 of 2) sorted by relevance
179 auto broadcast_op = op.tensor().getDefiningOp<BroadcastOp>(); in matchAndRewrite() local
743 static SmallVector<AffineMap, 2> getIndexingMaps(OpTy broadcast_op, in getIndexingMaps()781 mhlo::BroadcastInDimOp broadcast_op, Builder* b) { in getIndexingMaps()