Searched defs:sub_builder (Results 1 – 8 of 8) sorted by relevance
113 auto sub_builder = HloComputation::Builder("subcomputation"); in TEST_F() local
472 auto sub_builder = builder.CreateSubBuilder("ErrorAdd"); in TEST_F() local510 auto sub_builder = builder.CreateSubBuilder("power"); in TEST_F() local537 auto sub_builder = builder.CreateSubBuilder("power"); in TEST_F() local563 auto sub_builder = builder.CreateSubBuilder("power"); in TEST_F() local
595 HloComputation::Builder sub_builder("subcomp"); in TEST_F() local
142 auto sub_builder = HloComputation::Builder("max"); in TEST_F() local
1227 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); in TEST_F() local
2633 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); in TEST_F() local
1389 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); in TEST_F() local
2639 auto sub_builder = absl::make_unique<XlaBuilder>(computation_name); in CreateSubBuilder() local