Searched refs:WhileBodyFuncGraph (Results 1 – 4 of 4) sorted by relevance
42 WhileBodyFuncGraph = control_flow_v2_func_graphs.WhileBodyFuncGraph variable51 isinstance(graph, WhileBodyFuncGraph) or61 isinstance(graph, WhileBodyFuncGraph))
56 class WhileBodyFuncGraph(ControlFlowFuncGraph): class
200 func_graph=util.WhileBodyFuncGraph(880 class _WhileBodyGradFuncGraph(util.WhileBodyFuncGraph):
817 if (isinstance(val.op.graph, control_flow_v2_func_graphs.WhileBodyFuncGraph)