Searched refs:builder_cond (Results 1 – 1 of 1) sorted by relevance
572 auto builder_cond = HloComputation::Builder("cond"); in TEST_F() local573 auto cond_param = builder_cond.AddInstruction( in TEST_F()576 builder_cond.AddInstruction(CreateDot(shape, cond_param, cond_param)); in TEST_F()577 auto cond_root = builder_cond.AddInstruction(HloInstruction::CreateCompare( in TEST_F()579 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()581 builder_cond.AddInstruction( in TEST_F()584 builder_cond.AddInstruction(HloInstruction::CreateReshape( in TEST_F()586 builder_cond.AddInstruction( in TEST_F()590 auto cond = module->AddEmbeddedComputation(builder_cond.Build()); in TEST_F()632 auto builder_cond = HloComputation::Builder("cond"); in TEST_F() local[all …]