Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dxla_cluster_util.cc123 std::vector<ControlFlowInfo> control_flow_info; in CreateCycleDetectionGraph() local
124 TF_RETURN_IF_ERROR(BuildControlFlowInfo(graph, &control_flow_info)); in CreateCycleDetectionGraph()
164 control_flow_info[edge->dst()->id()].frame_name; in CreateCycleDetectionGraph()
172 control_flow_info[edge->src()->id()].frame_name; in CreateCycleDetectionGraph()
Dencapsulate_subgraphs_pass.cc1834 const std::vector<ControlFlowInfo>& control_flow_info, in AddDummyShapedNode() argument
1850 while (!control_flow_info[src_node->id()].frame_name.empty()) { in AddDummyShapedNode()
1855 control_flow_info[src_node->id()].frame_name); in AddDummyShapedNode()
1862 src_node = control_flow_info[src_node->id()].parent_frame; in AddDummyShapedNode()
1927 std::vector<ControlFlowInfo> control_flow_info; in DoStaticShapeInferenceForOutsideCompilationSend() local
1928 TF_RETURN_IF_ERROR(BuildControlFlowInfo(&graph_in, &control_flow_info)); in DoStaticShapeInferenceForOutsideCompilationSend()
1994 AddDummyShapedNode(src_node, src_port, control_flow_info, in DoStaticShapeInferenceForOutsideCompilationSend()