Searched refs:distribution_ (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dshape_partition_test.cc192 : generator_(rd_()), distribution_(1, 10) {}
196 int64 Rand() { return distribution_(generator_); }
200 std::uniform_int_distribution<int> distribution_; member in class:xla::cpu::__anon25566::RandomShapePartitionIteratorTest
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instruction.cc141 instruction->distribution_ = proto.distribution();
196 instruction->distribution_ = distribution;
1084 return distribution_;
1340 clone = CreateRng(shape, distribution_, new_operands);
2285 StrCat("distribution=", RandomDistributionToString(distribution_)));
2376 proto.set_distribution(distribution_);
H A Dhlo_instruction.h1468 RandomDistribution distribution_; member in class:xla::HloInstruction

Completed in 243 milliseconds