Searched refs:RemoveEmbeddedComputation (Results 1 – 7 of 7) sorted by relevance
86 return module->RemoveEmbeddedComputation(fused_computation); in Defuse()
124 return module->RemoveEmbeddedComputation(computation); in RecursivelyRemoveDeadComputation()
144 Status HloModule::RemoveEmbeddedComputation(HloComputation* to_remove) { in RemoveEmbeddedComputation() function in xla::HloModule938 TF_RETURN_IF_ERROR(RemoveEmbeddedComputation(computation)); in RemoveUnusedComputations()
95 Status RemoveEmbeddedComputation(HloComputation* to_remove);
415 TF_ASSERT_OK(module->RemoveEmbeddedComputation(reduction)); in TEST_F()
1319 TF_RETURN_IF_ERROR(module->RemoveEmbeddedComputation(temp_new_while_body)); in TryMergeInductionVariables()
1980 TF_CHECK_OK(parent()->parent()->RemoveEmbeddedComputation( in MergeFusionInstruction()2037 TF_CHECK_OK(GetModule()->RemoveEmbeddedComputation(computation_to_merge)); in MergeFusionInstructionIntoMultiOutput()