Searched refs:IsWhileNode (Results 1 – 13 of 13) sorted by relevance
165 } else if (n->IsWhileNode() && lower_control_flow(n)) { in Run()
59 } else if (node->IsWhileNode()) { in SetDeviceOrdinalAttributeForNode()
773 } else if (n->IsWhileNode()) { in PropagateConstIntoFunctionalNodes()804 if (!e->IsControlEdge() && e->dst()->IsWhileNode()) { in RewriteTensorListWithConstElement()818 if (e->src_output() == fwd_while_dst_input && e->dst()->IsWhileNode()) { in RewriteTensorListWithConstElement()
537 if (n->IsWhileNode()) { in RearrangeFunctionArguments()
418 } else if (n->IsWhileNode()) { in ResetDeviceOrdinalToPlaceholderValue()587 TF_RET_CHECK(n->IsWhileNode()); in PostprocessLiftedArgsForWhile()946 if (n->IsWhileNode()) { in PostprocessLiftedArgs()2122 } else if (n->IsWhileNode()) { in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
450 if (node.IsWhileNode() && in IsCompilableNode()
1018 bool has_functional_control_flow = node->IsWhileNode() || node->IsIfNode(); in BuildInitialClusterSet()
198 bool IsWhileNode() const { return class_ == NC_WHILE; } in IsWhileNode() function
635 if (!dst->IsWhileNode()) { in AddWhileInputHack()
1258 } else if (n->IsWhileNode()) { in InlineFunctionCalls()
3722 if (n->IsWhileNode() || n->IsIfNode() || IsFunctionCall(flib_def, *n)) { in LowerOutsideCompilationFunctionalNodes()3774 if (n->IsWhileNode()) { in LowerOutsideCompilationFunctionalNodes()3811 if (IsFunctionCall(flib_def, *node) || node->IsWhileNode() || in LowerOutsideCompilationFunctionalNodes()
2364 if (n->IsWhileNode()) { in LiftOutsideCompilationOnlyArgs()
988 if (node.IsWhileNode()) { in InferOutputType()2014 if (node.IsWhileNode()) { in ConvertNode()