Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/utils/
Dframe.cc65 if (IsEnter(*fanout.node)) { in InferFromGraphView()
99 if (IsEnter(*fanout.node)) { in InferFromGraphView()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op_test.cc78 ASSERT_FALSE(op->IsEnter()); in TEST()
99 if (op->IsEnter()) { in TEST()
179 ASSERT_FALSE(op->IsEnter()); in TEST()
195 if (op->IsEnter()) { in TEST()
Deval_const_tensor.cc204 if (IsEnter(current_node) || IsExit(current_node)) { in ExtractConstantSubgraph()
Dlower_if_while_test.cc269 ASSERT_FALSE(op->IsEnter()); in TEST()
Dexecutor.cc642 item->is_enter = IsEnter(n); in Initialize()
655 (IsEnter(n) || IsExit(n) || IsNextIteration(n)); in Initialize()
668 if (IsEnter(n)) { in Initialize()
1452 if (IsEnter(curr_node)) { in BuildControlFlowInfo()
/external/tensorflow/tensorflow/core/graph/
Dgraph.h149 bool IsEnter() const { return class_ == NC_ENTER; } in IsEnter() function
166 (IsSwitch() || IsMerge() || IsEnter() || IsExit() || in IsControlFlow()
738 inline bool IsEnter(const Node* node) { return node->IsEnter(); } in IsEnter() function
Dcontrol_flow.cc135 if (IsEnter(out)) { in BuildControlFlowInfo()
Dcontrol_flow_test.cc88 if (IsEnter(node)) { in TEST()
Dgraph_partition.cc365 if ((IsEnter(node) && !IsRefType(node->input_type(0))) || in OptimizeControlFlowColocation()
582 if (!node->IsEnter()) { in InputFrame()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc265 if (IsEnter(*invariant_node)) { in MoveInvariantNodes()
282 if (!IsConstant(*node) && !IsEnter(*node) && iter->second > 0) { in RevertInvariantNodes()
309 !IsEnter(*producer)) { in RevertInvariantNodes()
414 if (IsEnter(node) && node.attr().at("is_constant").b()) { in Optimize()
658 if (IsEnter(*node)) { in CheckForDeadFanout()
Dfunction_optimizer_test.cc278 EXPECT_TRUE(IsEnter(node)); in TEST_F()
Dconstant_folding.cc2331 if (IsEnter(*node) && node->input_size() > 0) { in MoveConstantsPastEnter()
Darithmetic_optimizer.cc3409 if (IsEnter(node) || IsExit(node)) { in CanDedup()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_cluster_util.cc155 if (edge->dst()->IsEnter() || edge->src()->IsExit()) { in CreateCycleDetectionGraph()
161 if (edge->dst()->IsEnter()) { in CreateCycleDetectionGraph()
/external/tensorflow/tensorflow/core/debug/
Ddebug_graph_utils.cc206 if (!src_node->IsEnter() && !src_node->IsNextIteration()) { in InsertNodes()
226 if (node->IsEnter()) { in DeparallelizeWhileLoops()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.cc210 bool IsEnter(const NodeDef& node) { in IsEnter() function
677 return IsEnter(node) || IsExit(node) || IsNextIteration(node); in ModifiesFrameInfo()
Dop_types.h67 bool IsEnter(const NodeDef& node);
/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc489 DCHECK(!n->IsEnter() && !n->IsNextIteration()) << n->DebugString(); in AddGradients()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc628 if (IsEnter(node)) { in FunctionalizeWhileLoop()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1867 if (IsEnter(*n)) { in UpdateShapes()
2066 if (IsEnter(*fanout.node)) { in InferStatically()