Home
last modified time | relevance | path

Searched refs:input_names (Results 1 – 25 of 27) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_attrs_process.cc35 std::vector<std::string> input_names = {"x"}; in SetAkgAttrsForFour2Five() local
37 AnfAlgo::SetNodeAttr("input_names", MakeValue(input_names), anf_node); in SetAkgAttrsForFour2Five()
52 std::vector<std::string> input_names = {"x"}; in SetAkgAttrsForFive2Four() local
54 AnfAlgo::SetNodeAttr("input_names", MakeValue(input_names), anf_node); in SetAkgAttrsForFive2Four()
79 std::vector<std::string> input_names = {"x", "dst_type"}; in SetAkgAttrsForCast() local
81 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), anf_node); in SetAkgAttrsForCast()
92 std::vector<std::string> input_names{"dy", "data", "mean"}; in SetAkgAttrsForBNGrad1() local
94 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), anf_node); in SetAkgAttrsForBNGrad1()
101 std::vector<std::string> input_names{"dgamma_red_hw", "dbeta_red_hw", "variance", "gamma"}; in SetAkgAttrsForBNGrad2() local
103 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), anf_node); in SetAkgAttrsForBNGrad2()
[all …]
/third_party/mindspore/mindspore/ccsrc/fl/server/
Dparameter_aggregator.cc263 const std::vector<std::string> &input_names = server_kernel->input_names(); in AssignMemory() local
265 if (input_names.size() != input_size_list.size()) { in AssignMemory()
267 << " input number is not matched: input_names size is " << input_names.size() in AssignMemory()
272 if (reuse_kernel_node_inputs_info.size() > input_names.size()) { in AssignMemory()
274 … << reuse_kernel_node_inputs_info.size() << ", but input_names size is " << input_names.size(); in AssignMemory()
278 for (size_t i = 0; i < input_names.size(); i++) { in AssignMemory()
279 const std::string &name = input_names[i]; in AssignMemory()
309 const std::vector<std::string> &input_names = aggr_kernel->input_names(); in GenerateAggregationKernelParams() local
310 …(void)std::transform(input_names.begin(), input_names.end(), std::back_inserter(aggr_params.inputs… in GenerateAggregationKernelParams()
333 const std::vector<std::string> &input_names = optimizer_kernel->input_names(); in GenerateOptimizerKernelParams() local
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Dsparse_softmax_cross_entropy_with_logits_unify_mindir.cc80 std::vector<std::string> input_names = {"indices", "depth", "on_value", "off_value"}; in CreateOneHot() local
82 one_hot_primitive->set_attr(kAttrInputNames, MakeValue(input_names)); in CreateOneHot()
171 std::vector<std::string> input_names = {"x", "axis"}; in CreateReduceMean() local
173 reduce_primitive->set_attr(kAttrInputNames, MakeValue(input_names)); in CreateReduceMean()
211 std::vector<std::string> input_names = {"x", "axis"}; in CreateExpandDims() local
213 expand_dims_primitive->set_attr(kAttrInputNames, MakeValue(input_names)); in CreateExpandDims()
234 std::vector<std::string> input_names = {"x"}; in CreateExpandDimsPynative() local
236 expand_dims_primitive->set_attr(kAttrInputNames, MakeValue(input_names)); in CreateExpandDimsPynative()
271 std::vector<std::string> input_names = {"x", "multiples"}; in CreateTile() local
273 tile_primitive->set_attr(kAttrInputNames, MakeValue(input_names)); in CreateTile()
[all …]
Dconv2d_unify_mindir.cc126 auto input_names = std::vector<std::string>{"x", "perm"}; in CreateTranspose()
128 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), transpose); in CreateTranspose()
236 auto input_names = std::vector<std::string>{"input_size", "filter", "dout"}; in SetConv2DBackpropInputAttrs()
237 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), depth_conv_backin); in SetConv2DBackpropInputAttrs()
248 auto input_names = std::vector<std::string>{"input", "filter_size", "dout"}; in SetConv2DBackpropFilterAttrs()
249 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), depth_conv_backfil); in SetConv2DBackpropFilterAttrs()
Davg_pool_grad_unify_mindir.cc214 auto input_names = std::vector<std::string>{"x_origin", "grad", "mean_matrix", "kernel_matrix"}; in Process()
215 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), avgpool_grad_vm); in Process()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dfusion_eltwise.cc408 std::vector<std::string> input_names; in CodegenCore() local
412 …input_names.push_back("in" + std::to_string(GetTensorIdx(reinterpret_cast<lite::Tensor *>(input.va… in CodegenCore()
416 input_names.push_back(var); in CodegenCore()
421 const std::string &var0 = input_names.at(0); in CodegenCore()
430 const std::string &var1 = input_names.at(1); in CodegenCore()
436 const std::string &var1 = input_names.at(1); in CodegenCore()
437 const std::string &var2 = input_names.at(2); in CodegenCore()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/
Dacl_model_options.cc67 void AclModelOptions::RenameInput(const std::vector<std::string> &input_names) { in RenameInput() argument
68 if (input_names.size() != input_shape_map_.size()) { in RenameInput()
78 input_shape_ += input_names[i] + ":" + s.substr(0, s.size() - 1) + ";"; in RenameInput()
Dacl_model.cc54 std::vector<std::string> input_names; in Build() local
64 input_names.push_back(name); in Build()
66 options->RenameInput(input_names); in Build()
/third_party/mindspore/tests/ut/cpp/pre_activate/common/
Drestore_abs_input_in_backed_infer_test.cc72 auto input_names = std::vector<std::string>{"a", "b", "c"}; in TEST_F()
75 prim_attr_test->AddAttr(kAttrInputNames, MakeValue(input_names)); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dadam_delta_cpu_kernel.cc99 std::vector<std::string> input_names = {"m", "v", "beta1_power", "beta2_power", "lr", in CheckParams() local
103 MS_LOG(EXCEPTION) << "Error input " << input_names[i] << " size!"; in CheckParams()
/third_party/mindspore/mindspore/lite/tools/converter/acl/
Dacl_pass.cc239 std::vector<std::string> input_names; in SetAclModelOptions() local
256 input_names.push_back(name); in SetAclModelOptions()
258 options_->RenameInput(input_names); in SetAclModelOptions()
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/
Daggregation_kernel.h74 const std::vector<std::string> &input_names() { return params_info_.inputs_names(); } in input_names() function
Doptimizer_kernel.h57 const std::vector<std::string> &input_names() { return params_info_.inputs_names(); } in input_names() function
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dgather_v2_ds_fission.cc106 auto input_names = AnfAlgo::GetNodeAttr<std::vector<std::string>>(origin_node, kAttrInputNames); in CreateGatherV2Ds() local
107 AnfAlgo::SetNodeAttr(kAttrInputNames, MakeValue(input_names), gather_v2); in CreateGatherV2Ds()
/third_party/mindspore/mindspore/lite/test/st/scripts/
Dbase_functions.sh147 input_names=`echo ${input_info} | sed 's/:/;/' | awk -F ';' '{print $2}'`
226 …MSLITE_BENCH_INPUT_NAMES=${input_names} ./benchmark --modelFile=${model_file} --inDataFile=${input…
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc828 const std::vector<std::string> &input_names, in ConvertInputNodes() argument
836 for (size_t j = 0; j < input_names.size(); j++) { in ConvertInputNodes()
837 std::string input_name = input_names[j]; // input may be produced by multi-outputs node in ConvertInputNodes()
952 std::vector<std::string> input_names; in ConvertOps() local
954 primitive_c = node_parser->Parse(node_def, tf_node_map, &input_names, &output_size); in ConvertOps()
963 primitive_c = node_parser_builtin->Parse(node_def, tf_node_map, &input_names, &output_size); in ConvertOps()
980 …status = ConvertInputNodes(node_def, input_names, tf_node_map, *anf_node_map, &inputs, &input_name… in ConvertOps()
Dtf_model_parser.h59 …ConvertInputNodes(const tensorflow::NodeDef &node_def, const std::vector<std::string> &input_names,
/third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/
Dmem_reuse_test.cc74 std::vector<std::string> input_names = {"x", "y"}; in CreateKernelGraph() local
76 ValuePtr input_names_v = MakeValue(input_names); in CreateKernelGraph()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc551 auto input_names = primitive->GetAttr(kAttrInputNames); in ConstInputToAttr() local
552 if (input_names == nullptr) { in ConstInputToAttr()
556 auto input_names_vec = GetValue<std::vector<std::string>>(input_names); in ConstInputToAttr()
731 auto input_names = primitive->GetAttr(kAttrInputNames); in RectifyAbstractFromRegAttr() local
732 if (input_names == nullptr) { in RectifyAbstractFromRegAttr()
735 auto input_names_vec = GetValue<std::vector<std::string>>(input_names); in RectifyAbstractFromRegAttr()
/third_party/skia/third_party/externals/tint/src/transform/
Dcanonicalize_entry_point_io.cc117 std::unordered_set<std::string> input_names; member
195 Symbol symbol = input_names.emplace(name).second in AddInput()
203 Symbol symbol = input_names.emplace(name).second in AddInput()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc661 std::vector<std::string> input_names = {"x"}; in CreateEndOfSequenceOP() local
662 ValuePtr input_names_v = MakeValue(input_names); in CreateEndOfSequenceOP()
693 std::vector<std::string> input_names = {"ref", "value"}; in CreateStreamAssignAddnOP() local
695 ValuePtr input_names_v = MakeValue(input_names); in CreateStreamAssignAddnOP()
978 std::vector<std::string> input_names = {"ref", "value"}; in CreateAssignAdd() local
980 ValuePtr input_names_v = MakeValue(input_names); in CreateAssignAdd()
1027 std::vector<std::string> input_names = {"ref", "value"}; in CreateAssign() local
1029 ValuePtr input_names_v = MakeValue(input_names); in CreateAssign()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dop_adapter.cc43 auto input_names = GetValue<const std::vector<std::string>>(value); in GenerateCustomOpInputMap() local
44 for (size_t i = 0; i < input_names.size(); ++i) { in GenerateCustomOpInputMap()
46 input_map[i + 1] = input_names[i]; in GenerateCustomOpInputMap()
47 op->CustomInputRegister(input_names[i]); in GenerateCustomOpInputMap()
/third_party/mindspore/mindspore/core/utils/
Dcheck_convert_utils.cc486 std::string input_names = ""; in CheckTensorTypeSame() local
488 (void)input_names.append(item.first); in CheckTensorTypeSame()
489 (void)input_names.append(", "); in CheckTensorTypeSame()
491 return CheckSubClass(input_names, check_type, check_list, prim_name); in CheckTensorTypeSame()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dub_pattern_fusion.cc74 std::vector<std::string> input_names; in CreateFusionOp() local
76 (void)input_names.emplace_back("input" + std::to_string(i)); in CreateFusionOp()
83 ValuePtr input_names_v = MakeValue(input_names); in CreateFusionOp()
/third_party/mindspore/mindspore/ccsrc/transform/express_ir/
Dmindir_exporter.cc485 std::vector<string> input_names; in BuildCNode() local
489 input_names.push_back(BuildInputNode(input, graph_proto)); in BuildCNode()
509 (void)std::for_each(input_names.begin(), input_names.end(), in BuildCNode()

12