Searched refs:rng_function (Results 1 – 1 of 1) sorted by relevance
591 HloComputation* rng_function = nullptr; in TEST_F() local605 rng_function = module->AddEmbeddedComputation(builder.Build()); in TEST_F()615 HloInstruction::CreateMap(constant->shape(), {constant}, rng_function)); in TEST_F()617 HloInstruction::CreateMap(constant->shape(), {constant}, rng_function)); in TEST_F()