Searched refs:kAttrOutputNames (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_attrs_process.cc | 82 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), anf_node); in SetAkgAttrsForCast()
95 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), anf_node); in SetAkgAttrsForBNGrad1()
104 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), anf_node); in SetAkgAttrsForBNGrad2()
127 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), anf_node); in SetAkgAttrsForBNGrad3()
136 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(fused_bn1_output_names), anf_node); in SetAkgAttrsForFusedBN1()
145 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(fused_bn2_output_names), anf_node); in SetAkgAttrsForFusedBN2()
154 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(fused_bn3_output_names), anf_node); in SetAkgAttrsForFusedBN3()
160 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(conv_bn1_output_names), anf_node); in SetAkgAttrsForConvBN1()
169 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(bn2_add_relu_output_names), anf_node); in SetAkgAttrsForBN2AddRelu()
177 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(bn2_output_names), anf_node); in SetAkgAttrsForBN2Relu()
|
| D | akg_kernel_json_generator.cc | 89 …td::set<std::string> black_list = {"IsFeatureMapInputList", "IsFeatureMapOutput", kAttrOutputNames, in ExcludeAttr()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | add_input_to_output.cc | 43 GetInputOrOutputNames(cnode, kAttrOutputNames, &output_names_vec); in AddOutputs()
71 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names_vec), cnode); in AddOutputs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 83 one_hot_primitive->set_attr(kAttrOutputNames, MakeValue(output_names)); in CreateOneHot()
174 reduce_primitive->set_attr(kAttrOutputNames, MakeValue(output_names)); in CreateReduceMean()
214 expand_dims_primitive->set_attr(kAttrOutputNames, MakeValue(output_names)); in CreateExpandDims()
237 expand_dims_primitive->set_attr(kAttrOutputNames, MakeValue(output_names)); in CreateExpandDimsPynative()
274 tile_primitive->set_attr(kAttrOutputNames, MakeValue(output_names)); in CreateTile()
321 real_div_primitive->set_attr(kAttrOutputNames, MakeValue(output_names)); in CreateRealDiv()
377 mul_primitive->set_attr(kAttrOutputNames, MakeValue(output_names)); in CreateMul()
|
| D | avg_pool_grad_unify_mindir.cc | 217 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), avgpool_grad_vm); in Process()
|
| D | conv2d_unify_mindir.cc | 129 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), transpose); in CreateTranspose()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | gather_v2_ds_fission.cc | 108 auto output_names = AnfAlgo::GetNodeAttr<std::vector<std::string>>(origin_node, kAttrOutputNames); in CreateGatherV2Ds()
109 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), gather_v2); in CreateGatherV2Ds()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 340 constexpr auto kAttrOutputNames = "output_names"; variable
|