| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | allgather_cpu_kernel.cc | 34 CHECK_KERNEL_INPUTS_NUM(input_num, kAllGatherInputsNum, kernel_name_); in InitKernel()
40 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kAllGatherInputsNum, kernel_name_); in Launch()
|
| D | update_cache_cpu_kernel.cc | 32 CHECK_KERNEL_INPUTS_NUM(input_num, kUpdateCacheInputsNum, kernel_name_); in InitKernel()
49 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kUpdateCacheInputsNum, kernel_name_); in Launch()
|
| D | random_cpu_kernel.cc | 142 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kUniformIntInputsNum, kernel_name_); in Launch()
146 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kUniformRealInputsNum, kernel_name_); in Launch()
|
| D | rmsprop_cpu_kernel.cc | 101 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kRMSPropInputsNum, kernel_name_); in Launch()
112 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kCenteredRMSPropInputsNum, kernel_name_); in Launch()
|
| D | debug_cpu_kernel.cc | 35 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kDebugInputsNum, kernel_name_); in Launch()
|
| D | equal_count_cpu_kernel.cc | 34 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kEqualCountInputsNum, kernel_name_); in Launch()
|
| D | apply_momentum_cpu_kernel.cc | 35 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kApplyMomentumInputsNum, kernel_name_); in Launch()
|
| D | l2loss_cpu_kernel.cc | 40 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kL2LossInputsNum, kernel_name_); in Launch()
|
| D | select_cpu_kernel.cc | 40 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kSelectInputsNum, kernel_name_); in Launch()
|
| D | adam_cpu_kernel.cc | 100 CHECK_KERNEL_INPUTS_NUM(input_num, kAdamInputsNum, kernel_name_); in InitKernel()
108 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kAdamInputsNum, kernel_name_); in Launch()
|
| D | reduce_scatter_cpu_kernel.cc | 53 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kReduceScatterInputsNum, kernel_name_); in Launch()
|
| D | reshape_cpu_kernel.cc | 37 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kReshapeInputsNum, kernel_name_); in Launch()
|
| D | unique_with_pad_cpu_kernel.cc | 30 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kUniqueWithPadInputsNum, kernel_name_); in Launch()
|
| D | hsigmoid_cpu_kernel.cc | 42 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kHSigmoidInputsNum, kernel_name_); in Launch()
|
| D | hswish_cpu_kernel.cc | 41 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kHSwishInputsNum, kernel_name_); in Launch()
|
| D | range_cpu_kernel.cc | 35 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kRangeInputsNum, kernel_name_); in Launch()
|
| D | expm1_cpu_kernel.cc | 39 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kExpm1InputsNum, kernel_name_); in Launch()
|
| D | hsigmoid_grad_cpu_kernel.cc | 42 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kHSigmoidGradInputsNum, kernel_name_); in Launch()
|
| D | cast_cpu_kernel.cc | 53 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kCastInputsNum, kernel_name_); in Launch()
|
| D | hswish_grad_cpu_kernel.cc | 42 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kHSwishGradInputsNum, kernel_name_); in Launch()
|
| D | smooth_l1_loss_grad_cpu_kernel.cc | 45 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kSmoothL1LossGradInputsNum, kernel_name_); in Launch()
|
| D | smooth_l1_loss_cpu_kernel.cc | 45 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kSmoothL1LossInputsNum, kernel_name_); in Launch()
|
| D | elu_grad_cpu_kernel.cc | 41 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kEleGradInputsNum, kernel_name_); in Launch()
|
| D | bias_add_grad_cpu_kernel.cc | 40 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kBiasAddGradInputsNum, kernel_name_); in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | log_softmax_cpu_kernel.cc | 52 CHECK_KERNEL_INPUTS_NUM(inputs.size(), kLogSoftmaxInputsNum, kernel_name_); in Launch()
|