Searched refs:cond_root (Results 1 – 2 of 2) sorted by relevance
1055 auto cond_root = cond_builder.AddInstruction( in TEST_F() local1091 schedule.set_sequence(condition, {cond_param, cond_root}); in TEST_F()
577 auto cond_root = builder_cond.AddInstruction(HloInstruction::CreateCompare( in TEST_F() local609 ShapeUtil::Equal(cond_root->shape(), ShapeUtil::MakeShape(PRED, {}))); in TEST_F()