Searched refs:reshard_node_def (Results 1 – 1 of 1) sorted by relevance
532 NodeDef reshard_node_def; in AddReshardOp() local533 reshard_node_def.set_name(graph->NewName(strings::StrCat( in AddReshardOp()535 reshard_node_def.set_op("TPUReshardVariables"); in AddReshardOp()537 &reshard_node_def); in AddReshardOp()538 Node* reshard_op_node = graph->AddNode(reshard_node_def, &status); in AddReshardOp()