| /external/tensorflow/tensorflow/compiler/jit/ |
| D | resource_operation_safety_analysis.cc | 258 return !edge.src()->IsNextIteration(); in ComputeIncompatibleResourceOperationPairs()
276 if (n->IsMerge() && e->src()->IsNextIteration()) { in ComputeIncompatibleResourceOperationPairs()
|
| D | shape_inference_helpers.cc | 34 if (e->src()->IsNextIteration()) { in Remove()
|
| D | deadness_analysis.cc | 988 if (backedge->src()->IsNextIteration()) { in CreateMultipleNextIterationInputsError()
1002 if (e->src()->IsNextIteration()) { in FindUniqueBackedge()
1110 if (!e->IsControlEdge() && e->src()->IsNextIteration()) { in HandleMerge()
1230 } else if (n->IsNextIteration()) { in HandleNode()
1263 if (IsNextIteration(e->src())) { in GetFrameBasedTopologicalOrder()
1293 if (IsNextIteration(curr_node) && IsMerge(out)) { in GetFrameBasedTopologicalOrder()
1449 if (n->IsNextIteration()) { in PopulateFrame()
|
| D | partially_decluster_pass.cc | 37 bool NotBackedge(const Edge& edge) { return !edge.src()->IsNextIteration(); } in NotBackedge()
|
| D | shape_inference.cc | 51 if (e.src->IsNextIteration() && e.dst->IsMerge()) { in PropagateShapes()
|
| D | xla_cluster_util.cc | 180 if (edge->src()->IsNextIteration()) { in CreateCycleDetectionGraph()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 161 bool IsNextIteration() const { return class_ == NC_NEXT_ITERATION; } in IsNextIteration() function
177 IsNextIteration()); in IsControlFlow()
816 inline bool IsNextIteration(const Node* n) { return n->IsNextIteration(); } in IsNextIteration() function
|
| D | validate.cc | 77 if (!e->IsControlEdge() && e->src()->IsNextIteration()) { in ValidateGraphHasNoCycle()
|
| D | graph_partition.cc | 54 inline bool IsNextIteration(const NodeDef& node_def) { in IsNextIteration() function
393 IsNextIteration(node)) { in OptimizeControlFlowColocation()
847 if (IsNextIteration(*ndef)) { in TopologicalSortNodesWithTimePriority()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op_test.cc | 96 ASSERT_FALSE(op->IsNextIteration()); in TEST()
129 if (op->IsNextIteration()) { in TEST()
442 ASSERT_FALSE(op->IsNextIteration()); in TEST()
469 if (op->IsNextIteration()) { in TEST()
|
| D | immutable_executor_state.cc | 192 (IsEnter(n) || IsExit(n) || IsNextIteration(n)); in Initialize()
196 item->is_next_iteration = IsNextIteration(n); in Initialize()
|
| D | lower_functional_ops_test.cc | 285 ASSERT_FALSE(op->IsNextIteration()); in TEST()
|
| D | graph_constructor.cc | 65 inline bool IsNextIteration(const NodeDef& node_def) { in IsNextIteration() function
697 if (IsNextIteration(node_def)) { in InitFromEdges()
1155 UpdatePendingCountAndReady(o, IsNextIteration(node_def)); in Convert()
1282 UpdatePendingCountAndReady(o, node->IsNextIteration()); in Convert()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | topological_sort.cc | 73 if (IsNextIteration(graph.node(input))) { in ComputeTopologicalOrder()
|
| D | scc.cc | 215 if (IsNextIteration(*node)) { in IdentifyLoops()
|
| D | graph_view.cc | 1465 if (IsNextIteration(graph_->node(curr_index)) && in SortTopologically()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 279 [](const Edge& edge) { return !edge.src()->IsNextIteration(); }); in BackwardsConstAnalysis()
|
| D | resource_util.cc | 55 n->IsNextIteration()); in IsControlFlowV1Node()
|
| D | functionalize_while.cc | 336 if (!IsNextIteration(arg.next_iteration)) { in FunctionalizeLoop()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | op_types.h | 127 bool IsNextIteration(const NodeDef& node);
|
| D | op_types.cc | 402 bool IsNextIteration(const NodeDef& node) { in IsNextIteration() function
768 return IsEnter(node) || IsExit(node) || IsNextIteration(node); in ModifiesFrameInfo()
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_graph_utils.cc | 239 if (!src_node->IsEnter() && !src_node->IsNextIteration()) { in InsertNodes()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 60 if (node->IsNextIteration()) return true; in IsExecuteNodeOrIdentityToExecuteNode()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradients.cc | 490 DCHECK(!n->IsEnter() && !n->IsNextIteration()) << n->DebugString(); in AddGradients()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1751 if (node.IsNextIteration()) { in CreateOperation()
2043 if (!edge.src->IsNextIteration() || !edge.dst->IsMerge()) { in AddBackedges()
|