| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_attrs_process.cc | 81 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), anf_node); in SetAkgAttrsForCast()
94 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), anf_node); in SetAkgAttrsForBNGrad1()
103 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), anf_node); in SetAkgAttrsForBNGrad2()
126 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), anf_node); in SetAkgAttrsForBNGrad3()
135 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(fused_bn1_input_names), anf_node); in SetAkgAttrsForFusedBN1()
144 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(fused_bn2_input_names), anf_node); in SetAkgAttrsForFusedBN2()
153 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(fused_bn3_input_names), anf_node); in SetAkgAttrsForFusedBN3()
167 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(bn2_add_relu_input_names), anf_node); in SetAkgAttrsForBN2AddRelu()
176 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(bn2_input_names), anf_node); in SetAkgAttrsForBN2Relu()
|
| D | akg_kernel_json_generator.h | 63 constexpr auto kAttrInputNames = "input_names"; variable
|
| D | akg_kernel_json_generator.cc | 90 kAttrInputNames}; in ExcludeAttr()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | change_axis_of_reduce_kernel.cc | 60 auto input_names_ptr = primitive->GetAttr(kAttrInputNames); in DynamicAttrUpdate()
65 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names_vec), node); in DynamicAttrUpdate()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | momentum_lossscale_fusion.cc | 92 …auto input_names_value = AnfAlgo::GetNodeAttr<std::vector<std::string>>(new_node, kAttrInputNames); in Process()
95 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names_value), new_node); in Process()
|
| D | add_input_to_output.cc | 41 GetInputOrOutputNames(cnode, kAttrInputNames, &input_names_vec); in AddOutputs()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/ |
| D | topk_split_test.cc | 110 …auto input_names_vec = AnfAlgo::GetNodeAttr<std::vector<std::string>>(topk_cnode, kAttrInputNames); in TEST_F()
115 input_names_vec = AnfAlgo::GetNodeAttr<std::vector<std::string>>(topk_cnode, kAttrInputNames); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 82 one_hot_primitive->set_attr(kAttrInputNames, MakeValue(input_names)); in CreateOneHot()
173 reduce_primitive->set_attr(kAttrInputNames, MakeValue(input_names)); in CreateReduceMean()
213 expand_dims_primitive->set_attr(kAttrInputNames, MakeValue(input_names)); in CreateExpandDims()
236 expand_dims_primitive->set_attr(kAttrInputNames, MakeValue(input_names)); in CreateExpandDimsPynative()
273 tile_primitive->set_attr(kAttrInputNames, MakeValue(input_names)); in CreateTile()
320 real_div_primitive->set_attr(kAttrInputNames, MakeValue(input_names)); in CreateRealDiv()
376 mul_primitive->set_attr(kAttrInputNames, MakeValue(input_names)); in CreateMul()
|
| D | conv2d_unify_mindir.cc | 128 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), transpose); in CreateTranspose()
225 AnfAlgo::CopyNodeAttr(kAttrInputNames, conv2d, depth_conv); in SetConv2DAttrs()
237 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), depth_conv_backin); in SetConv2DBackpropInputAttrs()
249 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), depth_conv_backfil); in SetConv2DBackpropFilterAttrs()
|
| D | slice_grad_unify_mindir.cc | 99 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(std::vector<std::string>{"x"}), pad); in Process()
|
| D | avg_pool_grad_unify_mindir.cc | 215 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), avgpool_grad_vm); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | gather_v2_ds_fission.cc | 106 auto input_names = AnfAlgo::GetNodeAttr<std::vector<std::string>>(origin_node, kAttrInputNames); in CreateGatherV2Ds()
107 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), gather_v2); in CreateGatherV2Ds()
|
| D | topk_split.cc | 95 auto input_names_vec = AnfAlgo::GetNodeAttr<std::vector<std::string>>(cnode, kAttrInputNames); in CheckInputNamesSize()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/common/ |
| D | restore_abs_input_in_backed_infer_test.cc | 75 prim_attr_test->AddAttr(kAttrInputNames, MakeValue(input_names)); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | helper.cc | 551 auto input_names = primitive->GetAttr(kAttrInputNames); in ConstInputToAttr()
731 auto input_names = primitive->GetAttr(kAttrInputNames); in RectifyAbstractFromRegAttr()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 337 constexpr auto kAttrInputNames = "input_names"; variable
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_select_graph_kernel.cc | 333 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue<std::vector<std::string>>({"x"}), trans_node); in UpdateEquivFormat()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 392 const auto &input_names_value = op_prim->GetAttr(kAttrInputNames); in RunOpConvertConstInputToAttr()
|