Searched refs:outer_builder (Results 1 – 3 of 3) sorted by relevance
718 auto outer_builder = HloComputation::Builder("outer_body"); in TEST_F() local719 auto outer_param = outer_builder.AddInstruction( in TEST_F()721 auto outer_element_0 = outer_builder.AddInstruction( in TEST_F()723 auto negate = outer_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()725 auto outer_element_1 = outer_builder.AddInstruction( in TEST_F()727 auto outer_tuple = outer_builder.AddInstruction( in TEST_F()729 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F()732 module_->AddEmbeddedComputation(outer_builder.Build()); in TEST_F()
330 auto outer_builder = HloComputation::Builder("OuterComputation"); in TEST_P() local331 auto outer_param0 = outer_builder.AddInstruction( in TEST_P()333 auto outer_param1 = outer_builder.AddInstruction( in TEST_P()336 auto nested_call = outer_builder.AddInstruction(HloInstruction::CreateCall( in TEST_P()339 module_->AddEmbeddedComputation(outer_builder.Build()); in TEST_P()687 auto outer_builder = HloComputation::Builder("outer_body"); in TEST_P() local688 auto outer_param = outer_builder.AddInstruction( in TEST_P()690 auto outer_element_0 = outer_builder.AddInstruction( in TEST_P()692 auto negate = outer_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_P()694 auto outer_element_1 = outer_builder.AddInstruction( in TEST_P()[all …]
115 XlaComputation CreateMinMaxComputation(XlaBuilder* outer_builder, in CreateMinMaxComputation() argument119 auto sub_builder = outer_builder->CreateSubBuilder("minmax_func"); in CreateMinMaxComputation()