Searched refs:SaveTaskInfo (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_parallel_build.h | 46 void SaveTaskInfo(int32_t task_id, const AnfNodePtr &anf_node, const std::string &json_name,
|
| D | tbe_kernel_parallel_build.cc | 88 build_manger->SaveTaskInfo(task_id, anf_node, json_name, input_size_list, output_size_list); in TbeOpParallelBuild()
126 void ParallelBuildManager::SaveTaskInfo(int32_t task_id, const mindspore::AnfNodePtr &anf_node, in SaveTaskInfo() function in mindspore::kernel::ParallelBuildManager
|
| D | ascend_kernel_compile.cc | 634 build_manager_->SaveTaskInfo(task_id, node, json_name, in_size_list, out_size_list); in AscendSingleOpCompile()
709 build_manager_->SaveTaskInfo(task_id, nullptr, json_name, input_size_list, output_size_list, in AscendFusionOpCompile()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_fusion.cc | 124 build_manger->SaveTaskInfo(task_id, nullptr, json_name, input_size_list, output_size_list, in KernelFusion()
|