| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_parallel_build.cc | 70 std::vector<size_t> input_size_list; in TbeOpParallelBuild() local
72 (void)TbeKernelBuild::GetIOSize(kernel_json, &input_size_list, &output_size_list); in TbeOpParallelBuild()
75 if (build_manger->SearchInCache(json_name, input_size_list, output_size_list, anf_node.get()) && in TbeOpParallelBuild()
81 build_manger->SaveSameOpInfo(anf_node, json_name, input_size_list, output_size_list); in TbeOpParallelBuild()
88 build_manger->SaveTaskInfo(task_id, anf_node, json_name, input_size_list, output_size_list); in TbeOpParallelBuild()
127 … const std::string &json_name, const std::vector<size_t> &input_size_list, in SaveTaskInfo() argument
138 task_info.input_size_list.assign(input_size_list.begin(), input_size_list.end()); in SaveTaskInfo()
189 …auto kernel_mod = GenKernelMod(task_iter->second.input_size_list, task_iter->second.output_size_li… in TaskFinishProcess()
219 const std::vector<size_t> &input_size_list, in SaveSameOpInfo() argument
225 task_info.input_size_list.assign(input_size_list.begin(), input_size_list.end()); in SaveSameOpInfo()
[all …]
|
| D | tbe_kernel_parallel_build.h | 37 std::vector<size_t> input_size_list; member
47 … const std::vector<size_t> &input_size_list, const std::vector<size_t> &output_size_list,
51 … const std::vector<size_t> &input_size_list, const std::vector<size_t> &output_size_list);
53 … const std::vector<size_t> &input_size_list, const std::vector<size_t> &output_size_list);
56 bool SearchInCache(const std::string &json_name, const std::vector<size_t> &input_size_list,
62 …KernelModPtr GenKernelMod(const std::vector<size_t> &input_size_list, const std::vector<size_t> &o…
|
| D | tbe_kernel_build.cc | 821 void GetInputSizeList(const nlohmann::json &input_json, std::vector<size_t> *input_size_list) { in GetInputSizeList() argument
822 MS_EXCEPTION_IF_NULL(input_size_list); in GetInputSizeList()
844 input_size_list->push_back(size_i); in GetInputSizeList()
879 … TbeKernelBuild::GetIOSize(const nlohmann::json &kernel_json, std::vector<size_t> *input_size_list, in GetIOSize() argument
881 if (input_size_list == nullptr || output_size_list == nullptr) { in GetIOSize()
885 input_size_list->clear(); in GetIOSize()
887 GetInputSizeList(kernel_json[kJOpInfo][kJInputs], input_size_list); in GetIOSize()
892 void GetRealInputSize(const nlohmann::json &input_json, std::vector<size_t> *input_size_list, size_… in GetRealInputSize() argument
908 input_size_list->push_back((*size_i)); in GetRealInputSize()
911 void GetInputSizeList2(const nlohmann::json &input_json, std::vector<size_t> *input_size_list) { in GetInputSizeList2() argument
[all …]
|
| D | tbe_kernel_build.h | 40 static bool GetIOSize2(const nlohmann::json &kernel_json, std::vector<size_t> *input_size_list,
42 static bool GetIOSize(const nlohmann::json &kernel_json, std::vector<size_t> *input_size_list,
49 … std::vector<size_t> *input_size_list, std::vector<size_t> *output_size_list);
50 …atic void CalInputSize(const nlohmann::json &fusion_op_list, std::vector<size_t> *input_size_list);
|
| D | ascend_kernel_compile.cc | 671 std::vector<size_t> input_size_list; in AscendFusionOpCompile() local
673 …f (!TbeKernelBuild::GetIOSize(fusion_op[kOpList], fusion_scope_iter.output_nodes, &input_size_list, in AscendFusionOpCompile()
681 … auto kernel_mod = build_manager_->GenKernelMod(input_size_list, output_size_list, kernel_pack); in AscendFusionOpCompile()
692 input_size_list, output_size_list); in AscendFusionOpCompile()
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 | 88 std::vector<size_t> input_size_list; in KernelFusion() local
90 …(!TbeKernelBuild::GetIOSize(fusion_op["op_list"], fusion_scope_iter.output_nodes, &input_size_list, in KernelFusion()
97 auto kernel_mod = build_manger->GenKernelMod(input_size_list, output_size_list, kernel_pack); in KernelFusion()
105 …SaveSameFusionOpInfo(fusion_scope_iter.scope_id, json_name, tbe::kProcessorAiCore, input_size_list, in KernelFusion()
124 build_manger->SaveTaskInfo(task_id, nullptr, json_name, input_size_list, output_size_list, in KernelFusion()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_build.cc | 43 std::vector<size_t> *input_size_list) { in SetIOIputSize() argument
45 MS_EXCEPTION_IF_NULL(input_size_list); in SetIOIputSize()
63 input_size_list->push_back(value.size()); in SetIOIputSize()
71 input_size_list->push_back(LongToSize(size_i)); in SetIOIputSize()
80 std::vector<size_t> input_size_list; in SetIOSize() local
85 if (!SetIOIputSize(anf_node, input_num, &input_size_list)) { in SetIOSize()
88 kernel_mod_ptr->SetInputSizeList(input_size_list); in SetIOSize()
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_lstm_op.py | 53 input_size_list = [input_size]
55 input_size_list.append(hidden_size * num_directions)
60 layers.append(nn.LSTMCell(input_size=input_size_list[i],
71 weight_size = (input_size_list[i] + hidden_size) * num_directions * hidden_size * 4
240 input_size_list = [input_size, hidden_size * num_directions]
244 weight_size = (input_size_list[i] + hidden_size) * num_directions * hidden_size * 4
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | parameter_aggregator.cc | 264 const std::vector<size_t> &input_size_list = server_kernel->GetInputSizeList(); in AssignMemory() local
265 if (input_names.size() != input_size_list.size()) { in AssignMemory()
268 << ", input_size_list size is " << input_size_list.size(); in AssignMemory()
295 auto input_addr = std::make_unique<char[]>(input_size_list[i]); in AssignMemory()
297 memory_register->RegisterArray(name, &input_addr, input_size_list[i]); in AssignMemory()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/ |
| D | akg_ascend_kernel_build.cc | 41 kernel_mod_ptr->SetInputSizeList(json_generator.input_size_list()); in AkgSetKernelMod()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/ |
| D | akg_gpu_kernel_build.cc | 43 kernel_mod_ptr->SetInputSizeList(json_generator.input_size_list()); in AkgSetKernelMod()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 380 const auto &input_size_list = kernel_mode->GetInputSizeList(); in FetchMemUnitSize() local
384 … size_t input_size = std::accumulate(input_size_list.begin(), input_size_list.end(), kZeroNumber); in FetchMemUnitSize()
394 if (input_num != input_size_list.size()) { in FetchMemUnitSize()
407 auto remove_size = kernel_ref_count_ptr->ref_count_ * input_size_list.at(i); in FetchMemUnitSize()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_generator.h | 105 const std::vector<size_t> &input_size_list() const { return input_size_list_; } in input_size_list() function
|