| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | tbe_json_utils.cc | 53 …onUtils::GetOutputsRealNum(const AnfNodePtr &anf_node, const std::vector<OpIOInfoPtr> &outputs_ptr, in GetOutputsRealNum() argument
57 for (const auto &output_ptr : outputs_ptr) { in GetOutputsRealNum()
59 if (outputs_ptr.size() > 1) { in GetOutputsRealNum()
61 … << " ] has " << outputs_ptr.size() << "outputs, however one of the outputs param_type is " in GetOutputsRealNum()
|
| D | single_tbe_json_creator.cc | 244 auto outputs_ptr = op_info_ptr->outputs_ptr(); in GenOutputsJson() local
245 if (!TbeJsonUtils::GetOutputsRealNum(anf_node, outputs_ptr, &outputs_tensor_num)) { in GenOutputsJson()
267 if (!AssignOutputsJson(anf_node, outputs_desc, outputs_tensor_num, outputs_ptr, &outputs_json)) { in GenOutputsJson()
278 const std::vector<OpIOInfoPtr> &outputs_ptr, in AssignOutputsJson() argument
284 auto output_ptr = outputs_ptr[i]; in AssignOutputsJson()
|
| D | single_tbe_json_creator.h | 41 … const std::vector<size_t> &outputs_tensor_num, const std::vector<OpIOInfoPtr> &outputs_ptr,
|
| D | tbe_json_utils.h | 112 …tic bool GetOutputsRealNum(const AnfNodePtr &anf_node, const std::vector<OpIOInfoPtr> &outputs_ptr,
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | cpu_kernel_factory.cc | 63 auto outputs_ptr = op_info->outputs_ptr(); in SetKernelAttrs() local
78 for (size_t j = 0; j < outputs_ptr.size(); j++) { in SetKernelAttrs()
79 auto output_dtypes = outputs_ptr[j]->dtypes(); in SetKernelAttrs()
80 auto output_formats = outputs_ptr[j]->formats(); in SetKernelAttrs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_select.cc | 98 const auto outputs_info = op_info.outputs_ptr(); in GetCommonPatternKernelInfo()
156 output_item.assign(op_info.outputs_ptr().size(), format); in GetAgnosticPatternKernelInfo()
480 op_info.outputs_ptr().size() != support_format.output_format[0].size()) { in CreateNewOpInfo()
483 << ", op info output size: " << op_info.outputs_ptr().size() in CreateNewOpInfo()
495 for (size_t j = 0; j < op_info.outputs_ptr().size(); ++j) { in CreateNewOpInfo()
496 auto output = op_info.outputs_ptr().at(j); in CreateNewOpInfo()
549 …if (op_info.inputs_ptr().size() != inputs.size() || op_info.outputs_ptr().size() != outputs.size()… in CreateNewOpInfo()
561 for (size_t i = 0; i < op_info.outputs_ptr().size(); ++i) { in CreateNewOpInfo()
563 …CreateNewOpIOInfo(*op_info.outputs_ptr().at(i), outputs.at(i).dtypes, outputs.at(i).formats, outpu… in CreateNewOpInfo()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | costmodel.h | 82 …: strategy_ptr(std::move(strategy)), inputs_ptr(std::move(inputs_)), outputs_ptr(std::move(outputs… in StrategyWithCost()
90 outputs_ptr(swc.outputs_ptr), in StrategyWithCost()
96 std::vector<TensorInfo> outputs_ptr; variable
|
| D | edge_costmodel.cc | 37 pre_op_output_.emplace_back(std::make_pair(swc->strategy_ptr, swc->outputs_ptr)); in InitEdgeCost()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/ |
| D | op_tiling_adapter.cc | 262 auto outputs_ptr = op_info_ptr->outputs_ptr(); in InitOpIoName() local
263 for (const auto &out_item : outputs_ptr) { in InitOpIoName()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/oplib/ |
| D | opinfo.h | 120 for (const auto &output : opinfo.outputs_ptr()) { in OpInfo()
139 std::vector<std::shared_ptr<OpIOInfo>> outputs_ptr() const { return outputs_ptr_; } in outputs_ptr() function
|
| D | oplib.cc | 310 …deDtypeFormat(dtype_format, op_io, op_info->inputs_ptr().size() + op_info->outputs_ptr().size())) { in DecodeInputOutput()
387 const auto &output_infos = op_info->outputs_ptr(); in GetRefInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.h | 97 return op_info->outputs_ptr().size(); in GetOpRegisteredOutputNum()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_build.cc | 466 auto outputs_ptr = op_info->outputs_ptr(); in GenTbeOutputsJson() local
467 return GenOutputDescJson(anf_node, outputs_ptr, outputs_json); in GenTbeOutputsJson()
472 …const std::vector<std::shared_ptr<mindspore::kernel::OpIOInfo>> &outputs_ptr, nlohmann::json *outp… in GenOutputDescJson() argument
478 for (const auto &output_ptr : outputs_ptr) { in GenOutputDescJson()
483 if (outputs_ptr.size() > 1) { in GenOutputDescJson()
|
| D | tbe_kernel_build.h | 118 … const std::vector<std::shared_ptr<OpIOInfo>> &outputs_ptr, nlohmann::json *outputs_json);
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | kernel_select_cpu.cc | 265 const auto &output_io_info = op_info->outputs_ptr(); in IsDynamicParamKernel()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | reshape_info.cc | 454 pre_out_tensor_infos = pre_stra_cost->outputs_ptr; in GenetateStrategyCosts()
|
| D | operator_info.cc | 1529 auto mem_cost = operator_cost()->GetMemoryCost(swc->inputs_ptr, swc->outputs_ptr); in CalculateMemoryCost()
1544 auto mem_cost = operator_cost()->GetMemoryCostForInference(swc->inputs_ptr, swc->outputs_ptr); in CalculateMemoryCostForInference()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | common_utils.cc | 436 std::vector<std::shared_ptr<OpIOInfo>> outputs = op_info_ptr->outputs_ptr(); in ParseMetadata()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_generator.cc | 231 auto outputs = op_info->outputs_ptr(); in CreateOutputDescJson()
|