Searched refs:unshard_node_def (Results 1 – 1 of 1) sorted by relevance
586 NodeDef unshard_node_def; in AddReshardOp() local587 unshard_node_def.set_name(graph->NewName(strings::StrCat( in AddReshardOp()589 unshard_node_def.set_op("TPUReshardVariables"); in AddReshardOp()591 &unshard_node_def); in AddReshardOp()592 Node* unshard_op_node = graph->AddNode(unshard_node_def, &status); in AddReshardOp()