Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op_test.cc92 ASSERT_FALSE(op->IsEnter()); in TEST()
116 if (op->IsEnter()) { in TEST()
206 ASSERT_FALSE(op->IsEnter()); in TEST()
225 if (op->IsEnter()) { in TEST()
Deval_const_tensor.cc207 if (IsEnter(current_node) || IsExit(current_node)) { in ExtractConstantSubgraph()
Dlower_functional_ops_test.cc282 ASSERT_FALSE(op->IsEnter()); in TEST()
Dexecutor.cc669 item->is_enter = IsEnter(n); in Initialize()
683 (IsEnter(n) || IsExit(n) || IsNextIteration(n)); in Initialize()
1531 if (IsEnter(curr_node)) { in BuildControlFlowInfo()
/external/tensorflow/tensorflow/core/grappler/utils/
Dframe.cc67 if (IsEnter(*fanout_node)) { in InferFromGraphViewT()
101 if (IsEnter(*fanout_node)) { in InferFromGraphViewT()
/external/tensorflow/tensorflow/core/graph/
Dgraph.h151 bool IsEnter() const { return class_ == NC_ENTER; } in IsEnter() function
168 (IsSwitch() || IsMerge() || IsEnter() || IsExit() || in IsControlFlow()
765 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.cc390 if ((IsEnter(node) && !IsRefType(node->input_type(0))) || in OptimizeControlFlowColocation()
607 if (!node->IsEnter()) { in InputFrame()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc263 if (IsEnter(*invariant_node)) { in MoveInvariantNodes()
280 if (!IsConstant(*node) && !IsEnter(*node) && iter->second > 0) { in RevertInvariantNodes()
307 !IsEnter(*producer)) { in RevertInvariantNodes()
414 if (IsEnter(node) && node.attr().at("is_constant").b()) { in Optimize()
666 if (IsEnter(*node)) { in CheckForDeadFanout()
Dfunction_optimizer.cc1099 return src->IsEnter() && CheckBoolAttr(src, "is_constant"); in AddStrictInputSemantics()
Darithmetic_optimizer.cc3446 if (IsEnter(node) || IsExit(node)) { in CanDedup()
Dconstant_folding.cc2347 if (!IsEnter(*node) || node->input_size() == 0 || in MoveConstantsPastEnter()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_util.cc74 if (IsEnter(node)) { in ExtractWhileLoopFrames()
Dresource_util.cc54 return (n->IsEnter() || n->IsExit() || n->IsSwitch() || n->IsMerge() || in IsControlFlowV1Node()
/external/tensorflow/tensorflow/core/debug/
Ddebug_graph_utils.cc239 if (!src_node->IsEnter() && !src_node->IsNextIteration()) { in InsertNodes()
259 if (node->IsEnter()) { in DeparallelizeWhileLoops()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_cluster_util.cc148 if (edge->dst()->IsEnter() || edge->src()->IsExit()) { in CreateCycleDetectionGraph()
154 if (edge->dst()->IsEnter()) { in CreateCycleDetectionGraph()
Ddeadness_analysis.cc882 return IsEnter(n) && control_flow_info_[n->id()].parent_frame->IsSource(); in IsRootEnter()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h73 bool IsEnter(const NodeDef& node);
Dop_types.cc233 bool IsEnter(const NodeDef& node) { in IsEnter() function
717 return IsEnter(node) || IsExit(node) || IsNextIteration(node); in ModifiesFrameInfo()
/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc489 DCHECK(!n->IsEnter() && !n->IsNextIteration()) << n->DebugString(); in AddGradients()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1985 if (IsEnter(*n)) { in UpdateShapes()
2186 if (IsEnter(*fanout.node)) { in InferStatically()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1425 if (node.IsEnter()) { in createOperation()