Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc75 std::vector<ControlFlowInfo> control_flow_info; in Run() local
76 TF_RETURN_IF_ERROR(BuildControlFlowInfo(g, &control_flow_info)); in Run()
82 const Node* frame = control_flow_info[n->id()].frame; in Run()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_cluster_util.cc116 std::vector<ControlFlowInfo> control_flow_info; in CreateCycleDetectionGraph() local
117 TF_RETURN_IF_ERROR(BuildControlFlowInfo(graph, &control_flow_info)); in CreateCycleDetectionGraph()
157 control_flow_info[edge->dst()->id()].frame_name; in CreateCycleDetectionGraph()
165 control_flow_info[edge->src()->id()].frame_name; in CreateCycleDetectionGraph()
Dencapsulate_subgraphs_pass.cc1001 const std::vector<ControlFlowInfo>& control_flow_info, in AddDummyShapedNode() argument
1017 while (!control_flow_info[src_node->id()].frame_name.empty()) { in AddDummyShapedNode()
1022 control_flow_info[src_node->id()].frame_name); in AddDummyShapedNode()
1029 src_node = control_flow_info[src_node->id()].parent_frame; in AddDummyShapedNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc1237 std::vector<ControlFlowInfo> control_flow_info; in InlineFunctionCalls() local
1238 TF_RETURN_IF_ERROR(BuildControlFlowInfo(graph.get(), &control_flow_info)); in InlineFunctionCalls()
1252 AddFrameForwardingControlEdge(control_flow_info, n, graph.get()); in InlineFunctionCalls()
1340 AddFrameForwardingControlEdge(control_flow_info, n, graph.get()); in InlineFunctionCalls()