Searched refs:distribution_ (Results 1 – 5 of 5) sorted by relevance
58 : distribution_(distribution), remaining_numbers_(0) {} in RandomSampleBuffer()63 results_ = (*distribution_)(random); in operator()79 Distribution* distribution_; variable
152 : generator_(rd_()), distribution_(1, 10) {} in RandomShapePartitionIteratorTest()156 int64 Rand() { return distribution_(generator_); } in Rand()160 std::uniform_int_distribution<int> distribution_; member in xla::cpu::__anon89ccd20e0111::RandomShapePartitionIteratorTest
45 return t + Clock::duration(distribution_(rand_)); in AddFuzz()55 std::uniform_int_distribution<int> distribution_{kFuzzLowerBoundClockTicks, member in openscreen::cast::__anone6f7ffcb0111::BandwidthEstimatorTest
1868 : HloInstruction(HloOpcode::kRng, shape), distribution_(distribution) { in HloRngInstruction()1876 proto.set_distribution(distribution_); in ToProto()1882 return {StrCat("distribution=", RandomDistributionToString(distribution_))}; in ExtraAttributesToStringImpl()1895 return distribution_ == casted_other.distribution_; in IdenticalSlowPath()1901 return absl::make_unique<HloRngInstruction>(shape, distribution_, in CloneWithNewOperandsImpl()
1021 RandomDistribution random_distribution() const { return distribution_; } in random_distribution()1040 RandomDistribution distribution_; variable