Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc550 const auto variable_edge = info.var_inputs[i]; in AddReshardOp() local
551 graph->AddEdge(variable_edge->src(), variable_edge->src_output(), in AddReshardOp()
599 const auto variable_edge = info.var_inputs[i]; in AddReshardOp() local
607 FindEnterNodeFromTPUExecuteNodeInput(variable_edge->src())); in AddReshardOp()