Searched refs:new_key_input (Results 1 – 1 of 1) sorted by relevance
555 const int new_key_input = info.var_inputs.size(); in AddReshardOp() local560 reshard_op_node, new_key_input); in AddReshardOp()582 const int format_state_input = new_key_input + 1; in AddReshardOp()615 graph->AddEdge(default_sharding_node, 0, unshard_op_node, new_key_input); in AddReshardOp()