Searched refs:HasPrimalAttr (Results 1 – 3 of 3) sorted by relevance
172 if (!cnode->HasPrimalAttr(PIPELINE_PARAM)) { in HandleReceiveParam()420 if (user_node->HasPrimalAttr(MICRO)) { in BroadCastMicroBatch()433 if (cnode->HasPrimalAttr(NEED_GRAD)) { in BroadCastNeedGrad()472 if (!IsInEndNodeBlackList(cur_node) && cur_node->HasPrimalAttr(NEED_GRAD)) { in GetPreNode()492 if (!cnode->HasPrimalAttr(MICRO)) { in LastStageEndNode()526 if (cnode->HasPrimalAttr(MICRO)) { in Micro()563 if (!cnode->HasPrimalAttr(MICRO)) { in HandleMicroBatch()595 if (cnode->HasPrimalAttr(kPrimalAttrForwardNodeName)) { in GetBorderNode()601 if (cnode->HasPrimalAttr(PIPELINE_END)) { in GetBorderNode()604 if (cnode->HasPrimalAttr(PIPELINE_BEGIN)) { in GetBorderNode()[all …]
324 …bool HasPrimalAttr(const std::string &name) const { return primal_attrs_.find(name) != attrs_.cend… in HasPrimalAttr() function
1929 if ((!StrategyFound(attrs) && !load_strategy_from_ckpt) && !cnode->HasPrimalAttr(STRATEGY)) { in ExtractInformation()1933 } else if (cnode->HasPrimalAttr(STRATEGY)) { in ExtractInformation()2574 if (!cnode->HasPrimalAttr(PIPELINE_PARAM)) { in ParallelCommunication()