Searched refs:broadcast_instr (Results 1 – 1 of 1) sorted by relevance
869 const HloInstruction* broadcast_instr = in ComputeArrayForElementwiseBinaryOp() local871 const HloInstruction* broadcast_const_operand = broadcast_instr->operand(0); in ComputeArrayForElementwiseBinaryOp()876 absl::Span<const int64> broadcast_dims = broadcast_instr->dimensions(); in ComputeArrayForElementwiseBinaryOp()892 broadcast_instr->shape().dimensions_size(), IndexComponent::Broadcasted); in ComputeArrayForElementwiseBinaryOp()