Searched refs:random_distribution (Results 1 – 8 of 8) sorted by relevance
56 std::uniform_int_distribution<Unsigned> random_distribution; in SemistaticMap() local59 hash_function.a = random_distribution(random_generator); in SemistaticMap()
301 switch (instruction->random_distribution()) { in HandleRng()325 RandomDistribution_Name(instruction->random_distribution())); in HandleRng()
1486 RandomDistribution random_distribution() const;
1447 switch (hlo->random_distribution()) { in ConvertValueForDistribution()1477 RandomDistribution_Name(hlo->random_distribution())); in ConvertValueForDistribution()
840 RandomDistribution random_distribution() const { return distribution_; } in random_distribution() function
2700 RandomDistribution distribution = random->random_distribution(); in HandleRng()2760 RandomDistribution distribution = random->random_distribution(); in HandleRng()
3363 RandomDistribution HloInstruction::random_distribution() const { in random_distribution() function in xla::HloInstruction3364 return Cast<HloRngInstruction>(this)->random_distribution(); in random_distribution()
483 "def random_distribution():\n",684 " feed = sample(random_distribution())\n",