Searched refs:kAttrParallelDimInfo (Results 1 – 3 of 3) sorted by relevance
644 SetNodeAttrSafely(kAttrParallelDimInfo, MakeValue<std::vector<size_t>>(info), fuse_nodes[i]); in SetFusedParallelOpAttrToReturnNode()651 SetNodeAttrSafely(kAttrParallelDimInfo, MakeValue<std::vector<size_t>>(info), inputs[j]); in SetFusedParallelOpAttrToReturnNode()656 SetNodeAttrSafely(kAttrParallelDimInfo, MakeValue<std::vector<size_t>>(info), out_node); in SetFusedParallelOpAttrToReturnNode()
752 if (AnfAlgo::HasNodeAttr(kAttrParallelDimInfo, tcnode)) { in GenParallelJson()753 auto info = AnfAlgo::GetNodeAttr<std::vector<size_t>>(tcnode, kAttrParallelDimInfo); in GenParallelJson()
463 constexpr auto kAttrParallelDimInfo = "parallel_dim_info"; variable