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