Searched refs:parent_computation (Results 1 – 1 of 1) sorted by relevance
544 auto parent_computation = all_reduce->parent(); in RewriteGraph() local549 TF_CHECK_OK(parent_computation->RemoveInstruction(all_reduce)); in RewriteGraph()574 auto divisor = parent_computation->AddInstruction( in RewriteGraph()576 auto division = parent_computation->AddInstruction( in RewriteGraph()