| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel_factory.cc | 42 if (kernel_info->GetInputNum() != iter_second->at(attr_index).first.GetInputSize()) { in CheckIOParam()
91 for (size_t input_index = 0; input_index < kernel_info->GetInputNum(); input_index++) { in ReducePrecision()
143 if (kernel_info->GetInputNum() > 0) { in GpuKernelAttrCheck()
147 for (size_t input_index = 0; input_index < kernel_info->GetInputNum(); input_index++) { in GpuKernelAttrCheck()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_query.cc | 45 kernel_build_info->GetInputNum() == input_tensor_num; in FilterInvalidKernelInfo()
60 << " cannot match the kernel's input size [" << kernel_info->GetInputNum() << "]"; in FilterInvalidKernelInfo()
|
| D | kernel_build_info.h | 93 size_t GetInputNum() const;
134 for (size_t index = 0; index < kernel_build_info->GetInputNum(); ++index) { in KernelBuildInfoBuilder()
|
| D | kernel_build_info.cc | 64 size_t KernelBuildInfo::GetInputNum() const { return inputs_format_.size(); } in GetInputNum() function in mindspore::kernel::KernelBuildInfo
104 for (size_t index = 0; index < GetInputNum(); ++index) { in ToString()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | merge_cast_to_op.cc | 38 for (size_t i = 0; i < candidate_kernel_info->GetInputNum(); i++) { in AlternativeKernelInfoForInput()
75 if (AnfAlgo::GetInputTensorNum(node) != kernel_info->GetInputNum()) { in CheckInputs()
79 for (size_t index = 0; index < kernel_info->GetInputNum(); ++index) { in CheckInputs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | layer_norm_beta_gamma_backprop_fusion.cc | 58 for (size_t i = 0; i < kernel_info->GetInputNum(); ++i) { in CheckKernelBuildInfo()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.cc | 43 size_t selected_input_num = selected_kernel_info->GetInputNum(); in CheckKernelInfo()
44 size_t alternative_input_num = alternative_kernel_info->GetInputNum(); in CheckKernelInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | cpu_kernel_factory.cc | 146 for (size_t i = 0; i < kernel_info.GetInputNum(); ++i) { in CPUKernelSingleAttrCheck()
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | full_quant_quantizer.h | 181 size_t GetInputNum() const { return data_pre_process_param_.calibrate_path_vector.size(); } in GetInputNum() function
|
| D | full_quant_quantizer.cc | 923 if (inputs.size() != calibrator_->GetInputNum()) { in DoInference()
925 << " calibrator count:" << calibrator_->GetInputNum(); in DoInference()
1190 if (inputs.size() != calibrator_->GetInputNum()) { in CollectDataFrequency()
1191 …OG(ERROR) << "model's input tensor cnt: " << inputs.size() << " != " << calibrator_->GetInputNum(); in CollectDataFrequency()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_select_ascend.cc | 61 for (size_t input_index = 0; input_index < kernel_build_info.GetInputNum(); ++input_index) { in MatchInferOutputDataType()
260 for (size_t input_index = 0; input_index < kernel_build_info->GetInputNum(); ++input_index) { in TagRaiseReduce()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 393 size_t in_num = AnfAlgo::GetInputNum(cnode); // include monads. in InsertTransOpForInput()
417 size_t in_num = AnfAlgo::GetInputNum(cnode); // include monads. in InsertCastForInput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | decrease_compute_precision.cc | 83 size_t in_num = AnfAlgo::GetInputNum(cnode); // include monads. in InsertCastForGraphKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | cpu_session.cc | 337 size_t input_num = kernel_build_Info->GetInputNum(); in KernelNotSupportException()
|
| D | anf_runtime_algorithm.h | 131 static size_t GetInputNum(const CNodePtr &cnode);
|
| D | anf_runtime_algorithm.cc | 575 size_t AnfRuntimeAlgorithm::GetInputNum(const CNodePtr &cnode) { in GetInputNum() function in mindspore::session::AnfRuntimeAlgorithm
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_scheduler.cc | 455 for (size_t i = 0; i < AnfAlgo::GetInputNum(kernel); ++i) { in Link()
|