Searched refs:reduce_comp_builder (Results 1 – 1 of 1) sorted by relevance
186 auto reduce_comp_builder = HloComputation::Builder("reduce_comp"); in TEST_F() local187 auto reduce_comp_param0 = reduce_comp_builder.AddInstruction( in TEST_F()189 auto reduce_comp_param1 = reduce_comp_builder.AddInstruction( in TEST_F()191 reduce_comp_builder.AddInstruction( in TEST_F()197 module->AddEmbeddedComputation(reduce_comp_builder.Build()); in TEST_F()