Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc571 Node* var_handle_node = graph->AddNode(var_handle_def, &status); in AddReshardOp() local
579 graph->AddControlEdge(loop_condition_node, var_handle_node); in AddReshardOp()
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()