Searched refs:var_handle_def (Results 1 – 1 of 1) sorted by relevance
565 NodeDef var_handle_def; in AddReshardOp() local566 var_handle_def.set_op("VarHandleOp"); in AddReshardOp()567 var_handle_def.set_name(graph->NewName(strings::StrCat( in AddReshardOp()569 AddNodeAttr("dtype", DT_STRING, &var_handle_def); in AddReshardOp()570 AddNodeAttr("shape", TensorShape({}), &var_handle_def); in AddReshardOp()571 Node* var_handle_node = graph->AddNode(var_handle_def, &status); in AddReshardOp()