Searched refs:default_sharding_node (Results 1 – 1 of 1) sorted by relevance
514 Node* default_sharding_node = graph->AddNode(default_sharding, &status); in AddReshardOp() local518 graph->AddControlEdge(loop_condition_node, default_sharding_node); in AddReshardOp()615 graph->AddEdge(default_sharding_node, 0, unshard_op_node, new_key_input); in AddReshardOp()