Home
last modified time | relevance | path

Searched defs:output_names (Results 1 – 21 of 21) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_attrs_process.cc36 std::vector<std::string> output_names = {"output"}; in SetAkgAttrsForFour2Five() local
53 std::vector<std::string> output_names = {"output"}; in SetAkgAttrsForFive2Four() local
80 std::vector<std::string> output_names = {"output"}; in SetAkgAttrsForCast() local
93 std::vector<std::string> output_names{"dgamma_red_hw", "dbeta_red_hw", "data_minus_mean"}; in SetAkgAttrsForBNGrad1() local
102 std::vector<std::string> output_names{"bn_scale", "bn_bias", "rs", "dgamma_dx", "dbeta_dx"}; in SetAkgAttrsForBNGrad2() local
125 std::vector<std::string> output_names{"dx"}; in SetAkgAttrsForBNGrad3() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Dsparse_softmax_cross_entropy_with_logits_unify_mindir.cc81 std::vector<std::string> output_names = {"output"}; in CreateOneHot() local
172 std::vector<std::string> output_names = {"y"}; in CreateReduceMean() local
212 std::vector<std::string> output_names = {"output"}; in CreateExpandDims() local
235 std::vector<std::string> output_names = {"output"}; in CreateExpandDimsPynative() local
272 std::vector<std::string> output_names = {"output"}; in CreateTile() local
319 std::vector<std::string> output_names = {"output"}; in CreateRealDiv() local
375 std::vector<std::string> output_names = {"output"}; in CreateMul() local
/third_party/mindspore/mindspore/ccsrc/fl/server/kernel/
Doptimizer_kernel.h59 const std::vector<std::string> &output_names() { return params_info_.outputs_names(); } in output_names() function
Daggregation_kernel.h76 const std::vector<std::string> &output_names() { return params_info_.outputs_names(); } in output_names() function
/third_party/ffmpeg/libavfilter/dnn/
Ddnn_backend_native.c262const char **output_names, uint32_t nb_output, AVFrame *out_frame, in execute_model_native()
376const char **output_names, uint32_t nb_output, AVFrame *out_frame) in ff_dnn_execute_model_native()
Ddnn_backend_tf.c714const char **output_names, uint32_t nb_output, AVFrame *out_frame, in execute_model_tf()
824const char **output_names, uint32_t nb_output, AVFrame *out_frame) in ff_dnn_execute_model_tf()
Ddnn_backend_openvino.c610const char **output_names, uint32_t nb_output, AVFrame *out_frame) in ff_dnn_execute_model_ov()
664const char **output_names, uint32_t nb_output, AVFrame *out_frame) in ff_dnn_execute_model_async_ov()
/third_party/mindspore/mindspore/lite/tools/converter/
Dconverter.cc132 auto output_names = ConverterContext::GetInstance()->GetGraphOutputTensorNames(); in Convert() local
Dconverter_context.h109 void SetGraphOutputTensorNames(const std::vector<std::string> &output_names) { in SetGraphOutputTensorNames()
/third_party/mindspore/mindspore/ccsrc/fl/server/
Dparameter_aggregator.cc317 const std::vector<std::string> &output_names = aggr_kernel->output_names(); in GenerateAggregationKernelParams() local
341 const std::vector<std::string> &output_names = optimizer_kernel->output_names(); in GenerateOptimizerKernelParams() local
/third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/src/
Dsession.cc113 Vector<String> output_names; in GetOutputTensorNames() local
/third_party/mindspore/mindspore/lite/micro/example/mnist_x86/src/
Dsession.cc114 Vector<String> output_names; in GetOutputTensorNames() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dgather_v2_ds_fission.cc108 auto output_names = AnfAlgo::GetNodeAttr<std::vector<std::string>>(origin_node, kAttrOutputNames); in CreateGatherV2Ds() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.cc664 std::vector<std::string> output_names = {"y"}; in CreateEndOfSequenceOP() local
694 std::vector<std::string> output_names = {"output"}; in CreateStreamAssignAddnOP() local
979 std::vector<std::string> output_names = {"output"}; in CreateAssignAdd() local
1028 std::vector<std::string> output_names = {"output"}; in CreateAssign() local
/third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/
Dmem_reuse_test.cc75 std::vector<std::string> output_names = {"output"}; in CreateKernelGraph() local
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_export.cc349 const std::vector<std::string> &output_names, const Model *model, in ExportNet()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dop_adapter.cc68 auto output_names = GetValue<const std::vector<std::string>>(value); in GenerateCustomOpOutputMap() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dub_pattern_fusion.cc78 std::vector<std::string> output_names; in CreateFusionOp() local
/third_party/mindspore/mindspore/lite/java/native/runtime/
Dlite_session.cpp207 auto output_names = lite_session_ptr->GetOutputTensorNames(); in Java_com_mindspore_lite_LiteSession_getOutputTensorNames() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_model_parser.cc528 std::vector<std::string> output_names; in ConvertGraphOutputs() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_model_parser.cc660 std::vector<std::string> output_names; in ConvertOnnxGraph() local