Searched refs:reshard_op_node (Results 1 – 1 of 1) sorted by relevance
538 Node* reshard_op_node = graph->AddNode(reshard_node_def, &status); in AddReshardOp() local541 reshard_op_node->set_assigned_device_name( in AddReshardOp()546 graph->AddControlEdge(before_loop_iteration_node, reshard_op_node); in AddReshardOp()547 graph->AddControlEdge(reshard_op_node, execute_node); in AddReshardOp()552 reshard_op_node, i); in AddReshardOp()560 reshard_op_node, new_key_input); in AddReshardOp()583 graph->AddEdge(var_handle_node, 0, reshard_op_node, format_state_input); in AddReshardOp()