Searched refs:ControlFlowInfo (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | control_flow.cc | 42 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()
|
| D | control_flow.h | 27 struct ControlFlowInfo { struct
55 Status BuildControlFlowInfo(const Graph* g, std::vector<ControlFlowInfo>* info,
|
| D | control_flow_test.cc | 60 std::vector<ControlFlowInfo> info; in TEST()
108 std::vector<ControlFlowInfo> info; in TEST()
133 std::vector<ControlFlowInfo> info; in TEST()
|
| D | graph_partition.cc | 114 std::vector<ControlFlowInfo> cf_info;
502 std::vector<ControlFlowInfo>* cf_info) { in AddControlFlowInfo()
507 const ControlFlowInfo& src_info = (*cf_info)[src->id()]; in AddControlFlowInfo()
508 ControlFlowInfo* info = &(*cf_info)[id]; in AddControlFlowInfo()
522 std::vector<ControlFlowInfo>* cf_info, in AddControlLoop()
526 const ControlFlowInfo& src_info = (*cf_info)[src->id()]; in AddControlLoop()
606 const std::vector<ControlFlowInfo>& cf_info) { in InputFrame()
616 const std::vector<ControlFlowInfo>& cf_info) { in OutputFrame()
639 std::vector<ControlFlowInfo>& cf_info = g_info->cf_info; in AddControlFlow()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | immutable_executor_state.h | 120 struct ControlFlowInfo { struct
126 ControlFlowInfo* cf_info); argument
127 void InitializePending(const Graph* graph, const ControlFlowInfo& cf_info);
|
| D | immutable_executor_state.cc | 88 ControlFlowInfo cf_info; in Initialize()
284 ControlFlowInfo* cf_info) { in BuildControlFlowInfo()
348 const ControlFlowInfo& cf_info) { in InitializePending()
|
| D | function_def_utils.cc | 68 std::vector<ControlFlowInfo> dummy; in FunctionDefToBodyHelper()
|
| D | accumulate_n_optimizer.cc | 75 std::vector<ControlFlowInfo> control_flow_info; in Run()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow_util.cc | 53 const std::vector<ControlFlowInfo>& cf_info, const Graph* graph, in ExtractWhileLoopFrames()
57 const ControlFlowInfo& cf = cf_info[node->id()]; in ExtractWhileLoopFrames()
|
| D | functionalize_control_flow_util.h | 78 const std::vector<ControlFlowInfo>& cf_info, const Graph* graph,
|
| D | functionalize_while.cc | 517 std::vector<ControlFlowInfo> cf_info; in FunctionalizeWhileLoop()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | deadness_analysis.cc | 892 std::vector<ControlFlowInfo> control_flow_info_;
1060 Status GetFullFrame(const Node* n, absl::Span<const ControlFlowInfo> cfi_infos, in GetFullFrame()
1063 for (const ControlFlowInfo* cfi_iter = &cfi_infos[n->id()]; !n->IsSource(); in GetFullFrame()
1079 Status GetRootFrame(const Node* n, absl::Span<const ControlFlowInfo> cfi_infos, in GetRootFrame()
1082 const ControlFlowInfo* cfi_iter = &cfi_infos[n->id()]; in GetRootFrame()
1251 const ControlFlowInfo& cf = control_flow_info_[node->id()]; in GetFrameBasedTopologicalOrder()
|
| D | xla_cluster_util.cc | 116 std::vector<ControlFlowInfo> control_flow_info; in CreateCycleDetectionGraph()
|
| D | encapsulate_subgraphs_pass.cc | 624 std::vector<ControlFlowInfo> dummy; in BuildFunctionDef()
1001 const std::vector<ControlFlowInfo>& control_flow_info, in AddDummyShapedNode()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 429 std::vector<ControlFlowInfo> cf_info; in DetectHostTrainingLoop()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 1149 void AddFrameForwardingControlEdge(const std::vector<ControlFlowInfo>& info, in AddFrameForwardingControlEdge()
1237 std::vector<ControlFlowInfo> control_flow_info; in InlineFunctionCalls()
|