Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc50 Node* execute_node; member
139 TF_RETURN_IF_ERROR(GetNodeAttr((*execute_node_info)[0].execute_node->attrs(), in ExtractExecuteNodeInfo()
155 auto execute = (*execute_node_info)[j].execute_node; in ExtractExecuteNodeInfo()
529 auto execute_node = info.execute_node; in AddReshardOp() local
542 execute_node->assigned_device_name()); in AddReshardOp()
547 graph->AddControlEdge(reshard_op_node, execute_node); in AddReshardOp()
558 FindEdgeConnecting(compile_node, execute_node); in AddReshardOp()
596 execute_node->assigned_device_name()); in AddReshardOp()
/external/tensorflow/tensorflow/core/common_runtime/eager/
DBUILD497 "execute_node.cc",
501 "execute_node.h",
617 "execute_node.cc",
621 "execute_node.h",