| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_query.cc | 39 size_t input_tensor_num = AnfAlgo::GetInputTensorNum(kernel_node); in FilterInvalidKernelInfo() local
43 …[output_tensor_num, input_tensor_num](const std::shared_ptr<kernel::KernelBuildInfo> &kernel_build… in FilterInvalidKernelInfo()
45 kernel_build_info->GetInputNum() == input_tensor_num; in FilterInvalidKernelInfo()
59 buffer << "Kernel node's input size [" << input_tensor_num << "]" in FilterInvalidKernelInfo()
66 << "input size : [" << input_tensor_num << "] can not match any kernelInfo !"; in FilterInvalidKernelInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/debug/ |
| D | print_gpu_kernel.h | 104 size_t input_tensor_num = AnfAlgo::GetInputTensorNum(kernel_node); in Init() local
105 input_flag_ = SetInputFlag(&string_pos_, input_tensor_num); in Init()
106 for (size_t i = 0; i < input_tensor_num; i++) { in Init()
191 std::vector<int64_t> SetInputFlag(std::vector<int64_t> *string_pos, size_t input_tensor_num) { in SetInputFlag() argument
194 std::vector<int64_t> res(string_pos->size() + input_tensor_num); in SetInputFlag()
197 if (res.size() == input_tensor_num) { in SetInputFlag()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | print_cpu_kernel.cc | 29 size_t input_tensor_num = AnfAlgo::GetInputTensorNum(kernel_node); in InitKernel() local
30 for (size_t i = 0; i < input_tensor_num; ++i) { in InitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_reuse.cc | 186 size_t input_tensor_num = AnfAlgo::GetInputTensorNum(kernel); in SetInputMap() local
187 for (size_t i = 0; i < input_tensor_num; ++i) { in SetInputMap()
191 if (input_tensor_num == 1) { in SetInputMap()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/ |
| D | task_generator.cc | 104 auto input_tensor_num = AnfAlgo::GetInputTensorNum(anf_node_ptr); in LaunchAddrCleanKernel() local
105 for (size_t i = 0; i < input_tensor_num; i++) { in LaunchAddrCleanKernel()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | anf.h | 312 ssize_t input_tensor_num() const { return input_tensor_num_; } in input_tensor_num() function
352 void set_input_tensor_num(ssize_t input_tensor_num) { input_tensor_num_ = input_tensor_num; } in set_input_tensor_num() argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_generator.cc | 190 size_t input_tensor_num = dyn_input_sizes.empty() ? 1 : IntToSize(dyn_input_sizes[i]); in CreateInputDescJson() local
192 for (size_t input_i = 0; input_i < input_tensor_num; input_i++) { in CreateInputDescJson()
687 size_t input_tensor_num = is_dynamic_input ? IntToSize(dyn_input_sizes[i]) : 1; in UpdateTensorName() local
688 for (size_t j = 0; j < input_tensor_num; ++j) { in UpdateTensorName()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | helper.h | 143 void CheckCNodeInputSize(const CNodePtr &cnode, size_t input_tensor_num);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_build.h | 119 bool GenInputList(const std::shared_ptr<AnfNode> &anf_node, size_t input_tensor_num,
|
| D | tbe_kernel_build.cc | 342 …eKernelJsonCreator::GenInputList(const std::shared_ptr<AnfNode> &anf_node, size_t input_tensor_num, in GenInputList() argument
354 for (size_t input_i = 0; input_i < input_tensor_num; input_i++) { in GenInputList()
439 size_t input_tensor_num; in GenTbeInputsJson() local
443 …if (!GetInputNameAndRealNum(anf_node, input_ptr, &dyn_input_index, &input_tensor_num, &op_input_na… in GenTbeInputsJson()
447 …if (!GenInputList(anf_node, input_tensor_num, input_ptr, &real_input_index, &op_input_name, &input… in GenTbeInputsJson()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/ |
| D | somas.cc | 514 auto input_tensor_num = AnfAlgo::GetInputTensorNum(kernel); in InitCommonNodeInputs() local
516 for (size_t i = 0; i < input_tensor_num; i++) { in InitCommonNodeInputs()
586 auto input_tensor_num = AnfAlgo::GetInputTensorNum(kernel); in InitAtomicCleanInputs() local
587 for (size_t i = 0; i < input_tensor_num; i++) { in InitAtomicCleanInputs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 591 ssize_t input_tensor_num = cnode->input_tensor_num(); in GetInputTensorNum() local
592 if (input_tensor_num >= 0) { in GetInputTensorNum()
593 return static_cast<size_t>(input_tensor_num); in GetInputTensorNum()
|
| D | ascend_session.cc | 1086 const auto input_tensor_num = AnfAlgo::GetInputTensorNum(cnode); in GetOpInputStubTensors() local
1087 for (size_t i = 1; i <= input_tensor_num; i += 1) { in GetOpInputStubTensors()
|
| D | session_basic.cc | 1439 const auto input_tensor_num = AnfAlgo::GetInputTensorNum(cnode); in GetOpInputTensors() local
1440 for (size_t i = 1; i <= input_tensor_num; i += 1) { in GetOpInputTensors()
|