Searched refs:broadcast_op (Results 1 – 1 of 1) sorted by relevance
412 static SmallVector<AffineMap, 2> getIndexingMaps(OpTy broadcast_op, in getIndexingMaps() argument415 broadcast_op.operand().getType().template cast<ShapedType>(); in getIndexingMaps()417 unsigned nloops = GetHloOpResultType<isLHLO>(broadcast_op).getRank(); in getIndexingMaps()421 unsigned num_prepended_dims = llvm::size(broadcast_op.broadcast_sizes()); in getIndexingMaps()450 mhlo::BroadcastInDimOp broadcast_op, Builder* b) { in getIndexingMaps() argument451 auto result_type = GetHloOpResultType<false>(broadcast_op); in getIndexingMaps()453 broadcast_op.operand().getType().template cast<ShapedType>(); in getIndexingMaps()466 if (broadcast_op.broadcast_dimensions()) { in getIndexingMaps()468 enumerate(broadcast_op.broadcast_dimensions().getIntValues())) { in getIndexingMaps()