Home
last modified time | relevance | path

Searched refs:output_data_desc (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_build.cc1165 nlohmann::json *output_data_desc) { in GenFusionOutputDescJson() argument
1167 MS_EXCEPTION_IF_NULL(output_data_desc); in GenFusionOutputDescJson()
1170 *output_data_desc = *output_desc; in GenFusionOutputDescJson()
1171 (*output_data_desc)[kJDtype] = (*output_desc)[kJDataType]; in GenFusionOutputDescJson()
1172 output_data_desc->erase(kJDataType); in GenFusionOutputDescJson()
1173 output_data_desc->erase(kJName); in GenFusionOutputDescJson()
1455 nlohmann::json output_data_desc; in GenFusionComputeOutputJson() local
1456 GenFusionOutputDescJson(cnode, i, desc_output_index[i], &output_desc, &output_data_desc); in GenFusionComputeOutputJson()
1457 output_data_desc_list->emplace_back(output_data_desc); in GenFusionComputeOutputJson()
1469 nlohmann::json output_data_desc; in GenFusionComputeOutputJson() local
[all …]
Dtbe_kernel_parallel_build.cc161 auto output_data_desc = GetJsonValue<nlohmann::json>(result, "op_params"); in PreTaskFinishProcess() local
165 AnfAlgo::SetOutputDataDesc(node, {output_data_desc}); in PreTaskFinishProcess()
Dtbe_kernel_build.h79 nlohmann::json *output_data_desc);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_build_info.h87 std::vector<nlohmann::json> output_data_desc() const { return output_data_desc_; } in output_data_desc() function
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc1347 return build_info->output_data_desc(); in GetOutputDataDesc()