Home
last modified time | relevance | path

Searched refs:expect_inputs_num (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.h148 #define CHECK_KERNEL_INPUTS_NUM(actual_inputs_num, expect_inputs_num, kernel_name) … argument
150 …if ((actual_inputs_num) != (expect_inputs_num)) { …
151 … MS_LOG(EXCEPTION) << (kernel_name) << " requires " << (expect_inputs_num) << " inputs, but got " \
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dbinary_cross_entropy_grad_kernel.cc74 …const size_t expect_inputs_num = weight_defined_ ? kBceGradInputsNumWithWeight : kBceGradInputsNum… in Launch() local
75 CHECK_KERNEL_INPUTS_NUM(inputs.size(), expect_inputs_num, kernel_name_); in Launch()
Dbinary_cross_entropy_cpu_kernel.cc95 …const size_t expect_inputs_num = weight_defined_ ? kBceInputsNumWithWeight : kBceInputsNumWithWeig… in Launch() local
96 CHECK_KERNEL_INPUTS_NUM(inputs.size(), expect_inputs_num, kernel_name_); in Launch()
Dslice_grad_cpu_kernel.cc95 size_t expect_inputs_num = in Launch() local
97 CHECK_KERNEL_INPUTS_NUM(inputs.size(), expect_inputs_num, kernel_name_); in Launch()