message: Ensure different distributions (e.g. from tf.random.normal and tf.random.uniform) are not considered identical.
diff: @@ -1818,7 +1818,8 @@ bool HloRngInstruction::IdenticalSlowPath(
  const HloInstruction& other,
  const std::function<bool(const HloComputation*, const HloComputation*)>&
  eq_computations) const {
  - return true;
  + const auto& casted_other = static_cast<const HloRngInstruction&>(other);
  + return distribution_ == casted_other.distribution_;
  }
  std::unique_ptr<HloInstruction> HloRngInstruction::CloneWithNewOperandsImpl(
  