Searched refs:RemoveEmbeddedComputation (Results 1 – 7 of 7) sorted by relevance
84 TF_RETURN_IF_ERROR(module->RemoveEmbeddedComputation(computation)); in Run()
87 return module->RemoveEmbeddedComputation(fused_computation); in Defuse()
80 Status RemoveEmbeddedComputation(HloComputation* to_remove);
291 TF_ASSERT_OK(module->RemoveEmbeddedComputation(reduction)); in TEST_F()
109 Status HloModule::RemoveEmbeddedComputation(HloComputation* to_remove) { in RemoveEmbeddedComputation() function in xla::HloModule
960 TF_RETURN_IF_ERROR(module->RemoveEmbeddedComputation(temp_new_while_body)); in TryMergeInductionVariables()
1265 TF_CHECK_OK(parent()->parent()->RemoveEmbeddedComputation( in MergeFusionInstruction()1313 TF_CHECK_OK(GetModule()->RemoveEmbeddedComputation(computation_to_merge)); in MergeFusionInstructionIntoMultiOutput()