Home
last modified time | relevance | path

Searched refs:real_input_num (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.cc300 …InputKernelBuilderInfo(const std::vector<std::shared_ptr<OpIOInfo>> &inputs, size_t real_input_num, in SetInputKernelBuilderInfo() argument
342 if (kernel_info_index < real_input_num) { in SetInputKernelBuilderInfo()
433 size_t real_input_num = AnfAlgo::GetInputTensorNum(kernel_node); in ParseMetadata() local
454 if (!SetInputKernelBuilderInfo(inputs, real_input_num, j, dyn_input_sizes, builder)) { in ParseMetadata()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dsingle_tbe_json_creator.cc119 size_t real_input_num = AnfAlgo::GetInputTensorNum(anf_node); in GenInputsJson() local
120 for (size_t i = 0; i < real_input_num; i++) { in GenInputsJson()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_build.cc352 size_t real_input_num = AnfAlgo::GetInputTensorNum(anf_node); in GenInputList() local
355 if (*real_input_index >= real_input_num) { in GenInputList()