Searched refs:outer_builder (Results 1 – 2 of 2) sorted by relevance
726 auto outer_builder = HloComputation::Builder("outer_body"); in TEST_F() local727 auto outer_param = outer_builder.AddInstruction( in TEST_F()729 auto outer_element_0 = outer_builder.AddInstruction( in TEST_F()731 auto negate = outer_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()733 auto outer_element_1 = outer_builder.AddInstruction( in TEST_F()735 auto outer_tuple = outer_builder.AddInstruction( in TEST_F()737 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()740 module_->AddEmbeddedComputation(outer_builder.Build()); in TEST_F()
423 auto outer_builder = HloComputation::Builder("OuterComputation"); in TEST_P() local424 auto outer_param0 = outer_builder.AddInstruction( in TEST_P()426 auto outer_param1 = outer_builder.AddInstruction( in TEST_P()429 auto nested_call = outer_builder.AddInstruction(HloInstruction::CreateCall( in TEST_P()432 module_->AddEmbeddedComputation(outer_builder.Build()); in TEST_P()686 auto outer_builder = HloComputation::Builder("outer_body"); in TEST_P() local687 auto outer_param = outer_builder.AddInstruction( in TEST_P()689 auto outer_element_0 = outer_builder.AddInstruction( in TEST_P()691 auto negate = outer_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_P()693 auto outer_element_1 = outer_builder.AddInstruction( in TEST_P()[all …]