Searched refs:rng0 (Results 1 – 4 of 4) sorted by relevance
94 let mut rng0 = StdRng::from_seed(seed); in test_stdrng_construction() localVariable95 let x0 = rng0.next_u64(); in test_stdrng_construction()97 let mut rng1 = StdRng::from_rng(rng0).unwrap(); in test_stdrng_construction()
161 HloInstruction *rng0, *rng1; in TEST_F() local168 rng0 = inst; in TEST_F()175 EXPECT_THAT(rng0->control_successors(), ElementsAre(rng1)); in TEST_F()176 EXPECT_THAT(rng1->control_predecessors(), ElementsAre(rng0)); in TEST_F()
81 auto rng0 = builder.AddInstruction(HloInstruction::CreateRng( variable89 builder.AddInstruction(HloInstruction::CreateReshape(root_shape, rng0));100 op::Add(op::Reshape(rng0), const1));105 op::Add(op::Reshape(rng0), const1));
2515 HloInstruction* rng0 = builder.AddInstruction( in TEST_F() local2520 HloInstruction::CreateTranspose(rng0->shape(), rng0, {1, 0})); in TEST_F()