Searched refs:cloned_computation (Results 1 – 2 of 2) sorted by relevance
203 HloComputation* cloned_computation = in TEST_F() local208 EXPECT_TRUE(cloned_computation->IsCustomCallComputation()); in TEST_F()209 EXPECT_EQ(cloned_computation->CustomCallInstruction(), cloned_custom_call); in TEST_F()271 HloComputation* cloned_computation = in TEST_F() local276 EXPECT_TRUE(cloned_computation->IsFusionComputation()); in TEST_F()277 EXPECT_EQ(cloned_computation->FusionInstruction(), cloned_fusion_instr); in TEST_F()
876 auto cloned_computation = entry_computation_->Clone(suffix, &context); in Clone() local877 module->AddEntryComputation(std::move(cloned_computation)); in Clone()