Searched refs:RemoveEmbeddedComputation (Results 1 – 7 of 7) sorted by relevance
107 TF_RETURN_IF_ERROR(module->RemoveEmbeddedComputation(computation)); in Run()
87 return module->RemoveEmbeddedComputation(fused_computation); in Defuse()
92 Status RemoveEmbeddedComputation(HloComputation* to_remove);
136 Status HloModule::RemoveEmbeddedComputation(HloComputation* to_remove) { in RemoveEmbeddedComputation() function in xla::HloModule753 TF_RETURN_IF_ERROR(RemoveEmbeddedComputation(computation)); in RemoveUnusedComputations()
289 TF_ASSERT_OK(module->RemoveEmbeddedComputation(reduction)); in TEST_F()
1172 TF_RETURN_IF_ERROR(module->RemoveEmbeddedComputation(temp_new_while_body)); in TryMergeInductionVariables()
1522 TF_CHECK_OK(parent()->parent()->RemoveEmbeddedComputation( in MergeFusionInstruction()1579 TF_CHECK_OK(GetModule()->RemoveEmbeddedComputation(computation_to_merge)); in MergeFusionInstructionIntoMultiOutput()