Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dcontrol_flow.cc42 const std::vector<ControlFlowInfo>& cf_info) { in ValidateControlFlowInfo()
45 const ControlFlowInfo& cf = cf_info[node->id()]; in ValidateControlFlowInfo()
86 Status BuildControlFlowInfo(const Graph* g, std::vector<ControlFlowInfo>* info, in BuildControlFlowInfo()
95 ControlFlowInfo& src_info = (*info)[src_node->id()]; in BuildControlFlowInfo()
105 const ControlFlowInfo& curr_info = (*info)[curr_node->id()]; in BuildControlFlowInfo()
112 const ControlFlowInfo& parent_info = (*info)[parent->id()]; in BuildControlFlowInfo()
121 ControlFlowInfo* out_info = &(*info)[out_id]; in BuildControlFlowInfo()
Dcontrol_flow.h27 struct ControlFlowInfo { struct
46 Status BuildControlFlowInfo(const Graph* g, std::vector<ControlFlowInfo>* info,
Dcontrol_flow_test.cc60 std::vector<ControlFlowInfo> info; in TEST()
108 std::vector<ControlFlowInfo> info; in TEST()
133 std::vector<ControlFlowInfo> info; in TEST()
Dgraph_partition.cc112 std::vector<ControlFlowInfo> cf_info;
475 std::vector<ControlFlowInfo>* cf_info) { in AddControlFlowInfo()
480 const ControlFlowInfo& src_info = (*cf_info)[src->id()]; in AddControlFlowInfo()
481 ControlFlowInfo* info = &(*cf_info)[id]; in AddControlFlowInfo()
495 std::vector<ControlFlowInfo>* cf_info, in AddControlLoop()
499 const ControlFlowInfo& src_info = (*cf_info)[src->id()]; in AddControlLoop()
579 const std::vector<ControlFlowInfo>& cf_info) { in InputFrame()
589 const std::vector<ControlFlowInfo>& cf_info) { in OutputFrame()
612 std::vector<ControlFlowInfo>& cf_info = g_info->cf_info; in AddControlFlow()
/external/tensorflow/tensorflow/compiler/jit/
Ddeadness_analysis.cc749 std::vector<ControlFlowInfo> control_flow_info_;
897 Status GetFullFrame(const Node* n, absl::Span<const ControlFlowInfo> cfi_infos, in GetFullFrame()
900 for (const ControlFlowInfo* cfi_iter = &cfi_infos[n->id()]; !n->IsSource(); in GetFullFrame()
Dxla_cluster_util.cc123 std::vector<ControlFlowInfo> control_flow_info; in CreateCycleDetectionGraph()
Dencapsulate_subgraphs_pass.cc1122 std::vector<ControlFlowInfo> dummy; in BuildFunctionDef()
1834 const std::vector<ControlFlowInfo>& control_flow_info, in AddDummyShapedNode()
1927 std::vector<ControlFlowInfo> control_flow_info; in DoStaticShapeInferenceForOutsideCompilationSend()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc599 std::vector<ControlFlowInfo> cf_info; in FunctionalizeWhileLoop()
611 const ControlFlowInfo& cf = cf_info[node->id()]; in FunctionalizeWhileLoop()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.cc320 struct ControlFlowInfo { struct in tensorflow::__anon675242740111::ExecutorImpl
356 ControlFlowInfo* cf_info);
357 void InitializePending(const Graph* graph, const ControlFlowInfo& cf_info);
602 ControlFlowInfo cf_info; in Initialize()
1426 ControlFlowInfo* cf_info) { in BuildControlFlowInfo()
1489 const ControlFlowInfo& cf_info) { in InitializePending()
Dfunction.cc2106 std::vector<ControlFlowInfo> dummy; in FunctionDefToBodyHelper()