Searched refs:kAttrFracZGroup (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | set_fracz_group_attr.cc | 54 AnfAlgo::SetNodeAttr(kAttrFracZGroup, MakeValue(groups), getitem_cnode); in GetOutputItem()
93 if (AnfAlgo::HasNodeAttr(kAttrFracZGroup, cnode)) { in HasFraczGroupAttrAndSet()
96 AnfAlgo::SetNodeAttr(kAttrFracZGroup, MakeValue(groups), cnode); in HasFraczGroupAttrAndSet()
125 AnfAlgo::SetNodeAttr(kAttrFracZGroup, MakeValue(groups), input); in GetCNodeNeighborFraczNodes()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | ub_pattern_fusion.cc | 90 if (AnfAlgo::HasNodeAttr(kAttrFracZGroup, cnode)) { in CreateFusionOp()
91 auto fracz_group = AnfAlgo::GetNodeAttr<int64_t>(node, kAttrFracZGroup); in CreateFusionOp()
92 fusion_op->set_attr(kAttrFracZGroup, MakeValue(fracz_group)); in CreateFusionOp()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 100 AnfAlgo::SetNodeAttr(kAttrFracZGroup, MakeValue(fz_group), out_trans); in SetGroupAttr()
103 AnfAlgo::SetNodeAttr(kAttrFracZGroup, MakeValue(fz_group), in_trans); in SetGroupAttr()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 390 constexpr auto kAttrFracZGroup = "fracz_group"; variable
|
| /third_party/mindspore/mindspore/ccsrc/common/ |
| D | trans.cc | 836 if (AnfAlgo::HasNodeAttr(kAttrFracZGroup, cnode)) { in GetAttrGroups()
847 return AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrFracZGroup); in GetAttrGroups()
|