| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | immutable_executor_state.cc | 104 if (IsSwitch(n) || IsMerge(n) || IsEnter(n) || IsExit(n)) { in Initialize()
157 item->is_enter = IsEnter(n); in Initialize()
192 (IsEnter(n) || IsExit(n) || IsNextIteration(n)); in Initialize()
310 if (IsEnter(curr_node)) { in BuildControlFlowInfo()
|
| D | lower_while_op_test.cc | 92 ASSERT_FALSE(op->IsEnter()); in TEST()
116 if (op->IsEnter()) { in TEST()
438 ASSERT_FALSE(op->IsEnter()); in TEST()
457 if (op->IsEnter()) { in TEST()
|
| D | lower_functional_ops_test.cc | 281 ASSERT_FALSE(op->IsEnter()); in TEST()
|
| D | eval_const_tensor.cc | 221 if (IsEnter(current_node) || IsExit(current_node)) { in ExtractConstantSubgraph()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | frame.cc | 67 if (IsEnter(*fanout_node)) { in InferFromGraphViewT()
101 if (IsEnter(*fanout_node)) { in InferFromGraphViewT()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 159 bool IsEnter() const { return class_ == NC_ENTER; } in IsEnter() function
176 (IsSwitch() || IsMerge() || IsEnter() || IsExit() || in IsControlFlow()
814 inline bool IsEnter(const Node* node) { return node->IsEnter(); } in IsEnter() function
|
| D | control_flow.cc | 135 if (IsEnter(out)) { in BuildControlFlowInfo()
|
| D | control_flow_test.cc | 88 if (IsEnter(node)) { in TEST()
|
| D | graph_partition.cc | 392 if ((IsEnter(node) && !IsRefType(node->input_type(0))) || in OptimizeControlFlowColocation()
609 if (!node->IsEnter()) { in InputFrame()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | loop_optimizer.cc | 263 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()
|
| D | common_subgraph_elimination.cc | 155 if (IsEnter(node) || IsExit(node)) { in CanDedup()
|
| D | function_optimizer.cc | 1115 return src->IsEnter() && CheckBoolAttr(src, "is_constant"); in AddStrictInputSemantics()
|
| D | constant_folding.cc | 2592 if (!IsEnter(*node) || node->input_size() == 0 || in MoveConstantsPastEnter()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow_util.cc | 75 if (IsEnter(node)) { in ExtractWhileLoopFrames()
|
| D | resource_util.cc | 54 return (n->IsEnter() || n->IsExit() || n->IsSwitch() || n->IsMerge() || in IsControlFlowV1Node()
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_graph_utils.cc | 239 if (!src_node->IsEnter() && !src_node->IsNextIteration()) { in InsertNodes()
259 if (node->IsEnter()) { in DeparallelizeWhileLoops()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_cluster_util.cc | 148 if (edge->dst()->IsEnter() || edge->src()->IsExit()) { in CreateCycleDetectionGraph()
154 if (edge->dst()->IsEnter()) { in CreateCycleDetectionGraph()
|
| D | deadness_analysis.cc | 882 return IsEnter(n) && control_flow_info_[n->id()].parent_frame->IsSource(); in IsRootEnter()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | op_types.h | 79 bool IsEnter(const NodeDef& node);
|
| D | op_types.cc | 260 bool IsEnter(const NodeDef& node) { in IsEnter() function
768 return IsEnter(node) || IsExit(node) || IsNextIteration(node); in ModifiesFrameInfo()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 86 if (node->IsEnter()) { in FindEnterNodeFromTPUExecuteNodeInput()
|
| D | distributed_tpu_rewrite_pass.cc | 2482 while (node->IsEnter() || node->IsSwitch()) { in FindVariableInputs()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradients.cc | 490 DCHECK(!n->IsEnter() && !n->IsNextIteration()) << n->DebugString(); in AddGradients()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 2300 if (IsEnter(*n)) { in UpdateShapes()
2501 if (IsEnter(*fanout.node)) { in InferStatically()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1768 if (node.IsEnter()) { in CreateOperation()
|