Home
last modified time | relevance | path

Searched refs:format_state_input (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc582 const int format_state_input = new_key_input + 1; in AddReshardOp() local
583 graph->AddEdge(var_handle_node, 0, reshard_op_node, format_state_input); in AddReshardOp()
618 graph->AddEdge(var_handle_node, 0, unshard_op_node, format_state_input); in AddReshardOp()