| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_attrs_process.cc | 36 std::vector<std::string> output_names = {"output"}; in SetAkgAttrsForFour2Five() local
38 AnfAlgo::SetNodeAttr("output_names", MakeValue(output_names), anf_node); in SetAkgAttrsForFour2Five()
53 std::vector<std::string> output_names = {"output"}; in SetAkgAttrsForFive2Four() local
55 AnfAlgo::SetNodeAttr("output_names", MakeValue(output_names), anf_node); in SetAkgAttrsForFive2Four()
80 std::vector<std::string> output_names = {"output"}; in SetAkgAttrsForCast() local
82 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), anf_node); in SetAkgAttrsForCast()
93 std::vector<std::string> output_names{"dgamma_red_hw", "dbeta_red_hw", "data_minus_mean"}; in SetAkgAttrsForBNGrad1() local
95 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), anf_node); in SetAkgAttrsForBNGrad1()
102 std::vector<std::string> output_names{"bn_scale", "bn_bias", "rs", "dgamma_dx", "dbeta_dx"}; in SetAkgAttrsForBNGrad2() local
104 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), anf_node); in SetAkgAttrsForBNGrad2()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 81 std::vector<std::string> output_names = {"output"}; in CreateOneHot() local
83 one_hot_primitive->set_attr(kAttrOutputNames, MakeValue(output_names)); in CreateOneHot()
172 std::vector<std::string> output_names = {"y"}; in CreateReduceMean() local
174 reduce_primitive->set_attr(kAttrOutputNames, MakeValue(output_names)); in CreateReduceMean()
212 std::vector<std::string> output_names = {"output"}; in CreateExpandDims() local
214 expand_dims_primitive->set_attr(kAttrOutputNames, MakeValue(output_names)); in CreateExpandDims()
235 std::vector<std::string> output_names = {"output"}; in CreateExpandDimsPynative() local
237 expand_dims_primitive->set_attr(kAttrOutputNames, MakeValue(output_names)); in CreateExpandDimsPynative()
272 std::vector<std::string> output_names = {"output"}; in CreateTile() local
274 tile_primitive->set_attr(kAttrOutputNames, MakeValue(output_names)); in CreateTile()
[all …]
|
| D | avg_pool_grad_unify_mindir.cc | 216 auto output_names = std::vector<std::string>{"output"}; in Process()
217 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), avgpool_grad_vm); in Process()
|
| D | conv2d_unify_mindir.cc | 127 auto output_names = std::vector<std::string>{"output"}; in CreateTranspose()
129 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), transpose); in CreateTranspose()
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | parameter_aggregator.cc | 317 const std::vector<std::string> &output_names = aggr_kernel->output_names(); in GenerateAggregationKernelParams() local
318 …(void)std::transform(output_names.begin(), output_names.end(), std::back_inserter(aggr_params.outp… in GenerateAggregationKernelParams()
341 const std::vector<std::string> &output_names = optimizer_kernel->output_names(); in GenerateOptimizerKernelParams() local
342 …(void)std::transform(output_names.begin(), output_names.end(), std::back_inserter(optimizer_params… in GenerateOptimizerKernelParams()
|
| /third_party/ffmpeg/libavfilter/dnn/ |
| D | dnn_backend_openvino.h | 35 … const char **output_names, uint32_t nb_output, AVFrame *out_frame);
37 … const char **output_names, uint32_t nb_output, AVFrame *out_frame);
|
| D | dnn_backend_native.c | 48 … const char **output_names, uint32_t nb_output, AVFrame *out_frame,
262 … const char **output_names, uint32_t nb_output, AVFrame *out_frame, in execute_model_native() argument
341 const char *output_name = output_names[i]; in execute_model_native()
376 … const char **output_names, uint32_t nb_output, AVFrame *out_frame) in ff_dnn_execute_model_native() argument
391 return execute_model_native(model, input_name, in_frame, output_names, nb_output, out_frame, 1); in ff_dnn_execute_model_native()
|
| D | dnn_backend_tf.h | 35 … const char **output_names, uint32_t nb_output, AVFrame *out_frame);
|
| D | dnn_backend_openvino.c | 610 … const char **output_names, uint32_t nb_output, AVFrame *out_frame) in ff_dnn_execute_model_ov() argument
641 if (init_model_ov(ov_model, input_name, output_names[0]) != DNN_SUCCESS) { in ff_dnn_execute_model_ov()
652 task.output_name = output_names[0]; in ff_dnn_execute_model_ov()
664 … const char **output_names, uint32_t nb_output, AVFrame *out_frame) in ff_dnn_execute_model_async_ov() argument
682 if (init_model_ov(ov_model, input_name, output_names[0]) != DNN_SUCCESS) { in ff_dnn_execute_model_async_ov()
699 task->output_name = output_names[0]; in ff_dnn_execute_model_async_ov()
|
| D | dnn_backend_tf.c | 66 … const char **output_names, uint32_t nb_output, AVFrame *out_frame,
714 … const char **output_names, uint32_t nb_output, AVFrame *out_frame, in execute_model_tf() argument
772 tf_outputs[i].oper = TF_GraphOperationByName(tf_model->graph, output_names[i]); in execute_model_tf()
776 av_log(ctx, AV_LOG_ERROR, "Could not find output \"%s\" in model\n", output_names[i]); \ in execute_model_tf()
824 … const char **output_names, uint32_t nb_output, AVFrame *out_frame) in ff_dnn_execute_model_tf() argument
839 return execute_model_tf(model, input_name, in_frame, output_names, nb_output, out_frame, 1); in ff_dnn_execute_model_tf()
|
| D | dnn_backend_native.h | 134 … const char **output_names, uint32_t nb_output, AVFrame *out_frame);
|
| /third_party/mindspore/mindspore/lite/tools/converter/ |
| D | converter.cc | 132 auto output_names = ConverterContext::GetInstance()->GetGraphOutputTensorNames(); in Convert() local
133 MS_ASSERT(output_names.size() == meta_graphT->outputIndex.size()); in Convert()
134 for (size_t idx = 0; idx < output_names.size(); idx++) { in Convert()
136 tensor->name = output_names.at(idx); in Convert()
|
| D | converter_context.h | 109 void SetGraphOutputTensorNames(const std::vector<std::string> &output_names) { in SetGraphOutputTensorNames() argument
110 graph_output_tensor_names_ = output_names; in SetGraphOutputTensorNames()
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/src/ |
| D | session.cc | 113 Vector<String> output_names; in GetOutputTensorNames() local
115 output_names.push_back(output->tensor_name()); in GetOutputTensorNames()
117 return output_names; in GetOutputTensorNames()
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_x86/src/ |
| D | session.cc | 114 Vector<String> output_names; in GetOutputTensorNames() local
116 output_names.push_back(output->tensor_name()); in GetOutputTensorNames()
118 return output_names; in GetOutputTensorNames()
|
| /third_party/ffmpeg/libavfilter/ |
| D | dnn_interface.h | 95 … const char **output_names, uint32_t nb_output, AVFrame *out_frame);
98 … const char **output_names, uint32_t nb_output, AVFrame *out_frame);
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | aggregation_kernel.h | 76 const std::vector<std::string> &output_names() { return params_info_.outputs_names(); } in output_names() function
|
| D | optimizer_kernel.h | 59 const std::vector<std::string> &output_names() { return params_info_.outputs_names(); } in output_names() function
|
| /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() local
109 AnfAlgo::SetNodeAttr(kAttrOutputNames, MakeValue(output_names), gather_v2); in CreateGatherV2Ds()
|
| /third_party/mindspore/mindspore/lite/src/train/ |
| D | train_export.cc | 349 const std::vector<std::string> &output_names, const Model *model, in ExportNet() argument
424 …if (std::find(output_names.begin(), output_names.end(), tensor->tensor_name()) != output_names.end… in ExportNet()
|
| D | train_export.h | 43 …onst std::vector<mindspore::lite::Tensor *> &tensors, const std::vector<std::string> &output_names,
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/ |
| D | mem_reuse_test.cc | 75 std::vector<std::string> output_names = {"output"}; in CreateKernelGraph() local
77 ValuePtr output_names_v = MakeValue(output_names); in CreateKernelGraph()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_adjust.cc | 664 std::vector<std::string> output_names = {"y"}; in CreateEndOfSequenceOP() local
665 ValuePtr output_names_v = MakeValue(output_names); in CreateEndOfSequenceOP()
694 std::vector<std::string> output_names = {"output"}; in CreateStreamAssignAddnOP() local
696 ValuePtr output_names_v = MakeValue(output_names); in CreateStreamAssignAddnOP()
979 std::vector<std::string> output_names = {"output"}; in CreateAssignAdd() local
981 ValuePtr output_names_v = MakeValue(output_names); in CreateAssignAdd()
1028 std::vector<std::string> output_names = {"output"}; in CreateAssign() local
1030 ValuePtr output_names_v = MakeValue(output_names); in CreateAssign()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | op_adapter.cc | 68 auto output_names = GetValue<const std::vector<std::string>>(value); in GenerateCustomOpOutputMap() local
69 for (size_t i = 0; i < output_names.size(); ++i) { in GenerateCustomOpOutputMap()
71 output_map[i] = output_names[i]; in GenerateCustomOpOutputMap()
72 op->CustomOutputRegister(output_names[i]); in GenerateCustomOpOutputMap()
|
| /third_party/ffmpeg/tools/python/ |
| D | convert_from_tensorflow.py | 65 self.output_names = []
424 if node.name in self.output_names:
474 self.output_names.append(node.name)
487 if name in self.output_names:
|