Searched refs:operand_to_broadcast (Results 1 – 2 of 2) sorted by relevance
3421 HloInstruction* operand_to_broadcast = in Visit() local3424 CHECK_EQ(ShapeUtil::Rank(operand_to_broadcast->shape()), in Visit()3434 operand_dim < ShapeUtil::Rank(operand_to_broadcast->shape()); in Visit()3439 operand_to_broadcast->shape().dimensions(operand_dim); in Visit()3443 operand_to_broadcast->shape().element_type(), output_dimensions); in Visit()3450 broadcast_shape, operand_to_broadcast, in Visit()3453 lhs = (lhs == operand_to_broadcast) ? broadcasted_operand : lhs; in Visit()3454 rhs = (rhs == operand_to_broadcast) ? broadcasted_operand : rhs; in Visit()
292 const Literal& operand_to_broadcast = in HandleBroadcast() local298 ShapeUtil::Rank(operand_to_broadcast.shape())) in HandleBroadcast()301 << ShapeUtil::Rank(operand_to_broadcast.shape()); in HandleBroadcast()306 operand_to_broadcast.shape().dimensions(i)); in HandleBroadcast()314 return operand_to_broadcast.Get<ReturnT>(broadcast_indices); in HandleBroadcast()