Searched refs:distribution_ (Results 1 – 3 of 3) sorted by relevance
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::__anon485538070111::RandomShapePartitionIteratorTest
1602 : HloInstruction(HloOpcode::kRng, shape), distribution_(distribution) { in HloRngInstruction()1610 proto.set_distribution(distribution_); in ToProto()1616 return {StrCat("distribution=", RandomDistributionToString(distribution_))}; in ExtraAttributesToStringImpl()1634 return absl::make_unique<HloRngInstruction>(shape, distribution_, in CloneWithNewOperandsImpl()
840 RandomDistribution random_distribution() const { return distribution_; } in random_distribution()859 RandomDistribution distribution_; variable