Home
last modified time | relevance | path

Searched refs:IsWhileNode (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dlower_functional_ops.cc169 } else if (n->IsWhileNode() && lower_control_flow(n)) { in Run()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dside_effect_util.cc68 } else if (node->IsWhileNode()) { in SetDeviceOrdinalAttributeForNode()
Dtf2xla_util.cc767 } else if (n->IsWhileNode()) { in PropagateConstIntoFunctionalNodes()
798 if (!e->IsControlEdge() && e->dst()->IsWhileNode()) { in RewriteTensorListWithConstElement()
812 if (e->src_output() == fwd_while_dst_input && e->dst()->IsWhileNode()) { in RewriteTensorListWithConstElement()
Drearrange_function_argument.cc530 if (n->IsWhileNode()) { in RearrangeFunctionArguments()
/external/tensorflow/tensorflow/compiler/jit/
Dcompilability_check_util.cc369 if (node.IsWhileNode() && in IsCompilableNode()
Dextract_outside_compilation_pass.cc416 } else if (n->IsWhileNode()) { in ResetDeviceOrdinalToPlaceholderValue()
571 TF_RET_CHECK(n->IsWhileNode()); in PostprocessLiftedArgsForWhile()
911 if (n->IsWhileNode()) { in PostprocessLiftedArgs()
2069 } else if (n->IsWhileNode()) { in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
Dmark_for_compilation_pass.cc993 bool has_functional_control_flow = node->IsWhileNode() || node->IsIfNode(); in BuildInitialClusterSet()
/external/tensorflow/tensorflow/core/graph/
Dgraph.h190 bool IsWhileNode() const { return class_ == NC_WHILE; } in IsWhileNode() function
Dgraph.cc610 if (!dst->IsWhileNode()) { in AddWhileInputHack()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc1226 } else if (n->IsWhileNode()) { in InlineFunctionCalls()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1640 if (node.IsWhileNode()) { in ConvertNode()