Home
last modified time | relevance | path

Searched refs:IsExit (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dimmutable_executor_state.cc104 if (IsSwitch(n) || IsMerge(n) || IsEnter(n) || IsExit(n)) { in Initialize()
188 item->is_exit = IsExit(n); in Initialize()
192 (IsEnter(n) || IsExit(n) || IsNextIteration(n)); in Initialize()
315 } else if (IsExit(curr_node)) { in BuildControlFlowInfo()
Dlower_while_op_test.cc93 ASSERT_FALSE(op->IsExit()); in TEST()
120 if (op->IsExit()) { in TEST()
439 ASSERT_FALSE(op->IsExit()); in TEST()
460 if (op->IsExit()) { in TEST()
Dlower_functional_ops_test.cc282 ASSERT_FALSE(op->IsExit()); in TEST()
Deval_const_tensor.cc221 if (IsEnter(current_node) || IsExit(current_node)) { in ExtractConstantSubgraph()
/external/tensorflow/tensorflow/core/grappler/utils/
Dframe.cc63 if (IsExit(*ready_node)) { in InferFromGraphViewT()
104 if (IsExit(*ready_node)) { in InferFromGraphViewT()
/external/tensorflow/tensorflow/core/graph/
Dgraph.h160 bool IsExit() const { return class_ == NC_EXIT; } in IsExit() function
176 (IsSwitch() || IsMerge() || IsEnter() || IsExit() || in IsControlFlow()
236 DCHECK(IsExit()); in set_while_ctx()
815 inline bool IsExit(const Node* node) { return node->IsExit(); } in IsExit() function
Dcontrol_flow.cc110 if (IsExit(curr_node)) { in BuildControlFlowInfo()
Dgraph_partition.cc382 } else if (IsExit(node)) { in OptimizeControlFlowColocation()
619 if (!node->IsExit()) { in OutputFrame()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc97 if (output_edge->IsControlEdge() || output_node->IsExit()) continue; in ResourceOnlyUsedForTPUExecuteInLoop()
288 if (e_switch->IsControlEdge() || !e_switch->dst()->IsExit()) continue; in FindLoopExitNodes()
352 if (out_node->IsExit()) { in AddNoOpAfterLastIteration()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_cluster_util.cc148 if (edge->dst()->IsEnter() || edge->src()->IsExit()) { in CreateCycleDetectionGraph()
162 if (edge->src()->IsExit()) { in CreateCycleDetectionGraph()
Ddeadness_analysis.cc886 return IsExit(n) && control_flow_info_[n->id()].parent_frame->IsSource(); in IsRootExit()
/external/tensorflow/tensorflow/cc/ops/
Dwhile_loop_test.cc100 ASSERT_TRUE(node->IsExit()) << "Output node " << i << ":\n" in TEST_F()
/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc402 DCHECK(exit_node->IsExit()); in ProcessWhileLoop()
484 if (n->IsExit()) { in AddGradients()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dcommon_subgraph_elimination.cc155 if (IsEnter(node) || IsExit(node)) { in CanDedup()
Dscoped_allocator_optimizer.cc221 *rewrite = IsConstant(*input) || IsExit(*input) || in MaybeRewriteInput()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dresource_util.cc54 return (n->IsEnter() || n->IsExit() || n->IsSwitch() || n->IsMerge() || in IsControlFlowV1Node()
Dfunctionalize_while.cc379 if (IsExit(edge->dst())) { in FunctionalizeLoop()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h81 bool IsExit(const NodeDef& node);
Dop_types.cc267 bool IsExit(const NodeDef& node) { in IsExit() function
768 return IsEnter(node) || IsExit(node) || IsNextIteration(node); in ModifiesFrameInfo()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1772 if (node.IsExit()) { in CreateOperation()