Searched refs:in_forward_flag (Results 1 – 5 of 5) sorted by relevance
80 return cnode->in_forward_flag(); in IsParallelCareNode()
444 if (!cnode->in_forward_flag()) { in HandleFullySplitParameters()
533 if (IsCommunicationOp(prim) && cnode->in_forward_flag()) { in FindCommunicationOp()1578 if (cnode->in_forward_flag() && !IsPrimitiveCNode(cnode, prim::kPrimReceive) && in ApplyParallelOptOnParam()1653 if (cnode->in_forward_flag()) users_count++; in SetSharedParameterFlag()2827 if (cnode->in_forward_flag()) { in HandleRootReshapeAndSaveStrategy()2983 if (!cnode->in_forward_flag()) { in HandleForwardMakeTupleAndMakeList()
293 bool in_forward_flag() const { return in_forward_flag_; } in in_forward_flag() function
114 if (IsPrimitiveCNode(cnode, prim::kPrimReceive) || !cnode->in_forward_flag()) { in InsertVirtualAssignAdd()