Home
last modified time | relevance | path

Searched defs:input_names (Results 1 – 22 of 22) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_attrs_process.cc35 std::vector<std::string> input_names = {"x"}; in SetAkgAttrsForFour2Five() local
52 std::vector<std::string> input_names = {"x"}; in SetAkgAttrsForFive2Four() local
79 std::vector<std::string> input_names = {"x", "dst_type"}; in SetAkgAttrsForCast() local
92 std::vector<std::string> input_names{"dy", "data", "mean"}; in SetAkgAttrsForBNGrad1() local
101 std::vector<std::string> input_names{"dgamma_red_hw", "dbeta_red_hw", "variance", "gamma"}; in SetAkgAttrsForBNGrad2() local
124 std::vector<std::string> input_names{"dy", "rs", "dgamma_dx", "dbeta_dx", "data_minus_mean"}; in SetAkgAttrsForBNGrad3() local
/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
171 std::vector<std::string> input_names = {"x", "axis"}; in CreateReduceMean() local
211 std::vector<std::string> input_names = {"x", "axis"}; in CreateExpandDims() local
234 std::vector<std::string> input_names = {"x"}; in CreateExpandDimsPynative() local
271 std::vector<std::string> input_names = {"x", "multiples"}; in CreateTile() local
318 std::vector<std::string> input_names = {"x", "y"}; in CreateRealDiv() local
374 std::vector<std::string> input_names = {"x", "y"}; in CreateMul() local
/third_party/mindspore/mindspore/ccsrc/fl/server/
Dparameter_aggregator.cc263 const std::vector<std::string> &input_names = server_kernel->input_names(); in AssignMemory() local
309 const std::vector<std::string> &input_names = aggr_kernel->input_names(); in GenerateAggregationKernelParams() local
333 const std::vector<std::string> &input_names = optimizer_kernel->input_names(); in GenerateOptimizerKernelParams() local
/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()
Dacl_model.cc54 std::vector<std::string> input_names; in Build() local
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/
Doptimizer_kernel.h57 const std::vector<std::string> &input_names() { return params_info_.inputs_names(); } in input_names() function
Daggregation_kernel.h74 const std::vector<std::string> &input_names() { return params_info_.inputs_names(); } in input_names() function
/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
/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
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc661 std::vector<std::string> input_names = {"x"}; in CreateEndOfSequenceOP() local
693 std::vector<std::string> input_names = {"ref", "value"}; in CreateStreamAssignAddnOP() local
978 std::vector<std::string> input_names = {"ref", "value"}; in CreateAssignAdd() local
1027 std::vector<std::string> input_names = {"ref", "value"}; in CreateAssign() local
/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
/third_party/mindspore/mindspore/lite/tools/converter/acl/
Dacl_pass.cc239 std::vector<std::string> input_names; in SetAclModelOptions() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dfusion_eltwise.cc408 std::vector<std::string> input_names; in CodegenCore() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc828 const std::vector<std::string> &input_names, in ConvertInputNodes()
952 std::vector<std::string> input_names; in ConvertOps() local
/third_party/skia/third_party/externals/tint/src/transform/
Dcanonicalize_entry_point_io.cc117 std::unordered_set<std::string> input_names; member
/third_party/mindspore/mindspore/lite/tools/benchmark/
Dbenchmark.cc801 auto input_names = StrSplit(bench_inputs, std::string(DELIM_COMMA)); in CheckInputNames() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc551 auto input_names = primitive->GetAttr(kAttrInputNames); in ConstInputToAttr() local
731 auto input_names = primitive->GetAttr(kAttrInputNames); in RectifyAbstractFromRegAttr() local
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dop_adapter.cc43 auto input_names = GetValue<const std::vector<std::string>>(value); in GenerateCustomOpInputMap() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dub_pattern_fusion.cc74 std::vector<std::string> input_names; in CreateFusionOp() local
/third_party/mindspore/mindspore/ccsrc/transform/express_ir/
Dmindir_exporter.cc485 std::vector<string> input_names; in BuildCNode() local
/third_party/mindspore/mindspore/core/utils/
Dcheck_convert_utils.cc486 std::string input_names = ""; in CheckTensorTypeSame() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc1787 std::vector<std::string> input_names; in GetModelOutputsInfo() local