Searched refs:reshaped_operand (Results 1 – 2 of 2) sorted by relevance
1419 HloInstruction* reshaped_operand = adder(HloInstruction::CreateReshape( in CreateBroadcastSequence() local1423 reshaped_operand->set_metadata(operand->metadata()); in CreateBroadcastSequence()1425 reshaped_operand->set_sharding(operand->sharding()); in CreateBroadcastSequence()1427 reshaped_operand->set_frontend_attributes(operand->frontend_attributes()); in CreateBroadcastSequence()1430 broadcast_shape, reshaped_operand, broadcast_dimensions); in CreateBroadcastSequence()
609 XlaOp reshaped_operand, in AddBroadcastSequence()612 return InDimBroadcast(broadcast_shape, reshaped_operand, in AddBroadcastSequence()