Searched refs:parallel_name (Results 1 – 3 of 3) sorted by relevance
69 auto parallel_name = IsMultiParallelConvNode(node); in Process() local70 if (parallel_name.empty()) { in Process()85 if (strategys_.find(parallel_name) == strategys_.end()) { in Process()86 MS_LOG(ERROR) << "Find " << parallel_name << " strategy failed"; in Process()90 …std::make_shared<MultiNodeSplitProxy>(strategys_.at(parallel_name), primitive_type_, fmk_type_, nu… in Process()
47 bool ParallelPass::SetParallelOpName(const AnfNodePtr &node, std::string *parallel_name) { in SetParallelOpName() argument48 MS_ASSERT(node != nullptr && parallel_name != nullptr); in SetParallelOpName()60 if (split_strategys_.find(*parallel_name) == split_strategys_.end()) { in SetParallelOpName()61 *parallel_name = type_name_; in SetParallelOpName()65 if (split_strategys_.find(*parallel_name) == split_strategys_.end()) { in SetParallelOpName()66 MS_LOG(DEBUG) << *parallel_name << " : No split strategy for the current CNode."; in SetParallelOpName()
45 bool SetParallelOpName(const AnfNodePtr &node, std::string *parallel_name);