Searched refs:after_unshard (Results 1 – 1 of 1) sorted by relevance
521 NodeDef after_unshard; in AddReshardOp() local522 after_unshard.set_op("NoOp"); in AddReshardOp()523 after_unshard.set_name(graph->NewName(strings::StrCat( in AddReshardOp()525 auto after_unshard_node = graph->AddNode(after_unshard, &status); in AddReshardOp()