| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | generator_cpu_kernel.cc | 84 constexpr size_t kInputNum = 4; in StepCompute() local
108 constexpr size_t kInputNum = 3; in SeedCompute() local
129 constexpr size_t kInputNum = 3; in GetStateCompute() local
147 constexpr size_t kInputNum = 4; in SetStateCompute() local
181 constexpr size_t kInputNum = 4; in ManualSeedCompute() local
204 constexpr size_t kInputNum = 3; in InitialSeedCompute() local
|
| D | hsv_to_rgb_cpu_kernel.h | 52 const size_t kInputNum = 1; variable
|
| D | lu_unpack_cpu_kernel.h | 28 constexpr size_t kInputNum = 2; variable
|
| D | lu_solve_cpu_kernel.h | 27 constexpr size_t kInputNum = 3; variable
|
| D | adaptive_max_pool_3d_grad_cpu_kernel.h | 58 const size_t kInputNum = 3; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | sigmoid_cross_entropy_with_logits.cc | 45 const int64_t kInputNum = 2; in InferShape() local
71 const int64_t kInputNum = 2; in InferType() local
|
| D | softmax_cross_entropy_with_logits.cc | 37 const int64_t kInputNum = 2; in InferShape() local
72 const int64_t kInputNum = 2; in InferType() local
|
| D | quant_dtype_cast.cc | 65 const int64_t kInputNum = 1; in InferShape() local
77 const int64_t kInputNum = 1; in InferType() local
|
| D | add_v2.cc | 38 const int64_t kInputNum = 2; in AddV2InferShape() local
53 constexpr int64_t kInputNum = 2; in AddV2InferType() local
|
| D | apply_rms_prop.cc | 52 const int64_t kInputNum = 5; in InferShape() local
100 const int64_t kInputNum = 8; in InferType() local
|
| D | map_cache_idx.cc | 47 constexpr int64_t kInputNum = 5; in MapCacheIdxInferShape() local
74 constexpr int64_t kInputNum = 5; in MapCacheIdxInferType() local
|
| D | apply_adagrad.cc | 55 const int64_t kInputNum = 4; in InferShape() local
100 const int64_t kInputNum = 4; in InferType() local
|
| D | sparse_matrix_softmax.cc | 48 constexpr size_t kInputNum = 5; in SparseMatrixSoftmaxInferShape() local
76 constexpr size_t kInputNum = 5; in SparseMatrixSoftmaxInferType() local
|
| D | ps_roi_pooling.cc | 46 const int64_t kInputNum = 2; in PSROIPoolingInferShape() local
116 const int64_t kInputNum = 2; in PSROIPoolingInferType() local
|
| D | apply_ftrl.cc | 51 const int64_t kInputNum = 8; in InferShape() local
112 const int64_t kInputNum = 8; in InferType() local
|
| D | apply_proximal_adagrad.cc | 49 const int64_t kInputNum = 6; in ApplyProximalAdagradInferShape() local
89 const int64_t kInputNum = 6; in ApplyProximalAdagradInferType() local
|
| D | apply_centered_rms_prop.cc | 47 const int64_t kInputNum = 9; in ApplyCenteredRMSPropInferShape() local
109 const int64_t kInputNum = 9; in ApplyCenteredRMSPropInferType() local
|
| D | roi_align.cc | 52 constexpr int64_t kInputNum = 2; in InferShape() local
99 constexpr int64_t kInputNum = 2; in InferType() local
|
| D | fmin.cc | 51 const int64_t kInputNum = 2; in InferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | sspaddmm.h | 36 const uint32_t kInputNum = 9; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | lu_unpack_gpu_kernel.h | 35 constexpr size_t kInputNum = 2; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | sigmoid_cross_entropy_with_logits_grad.cc | 51 const int64_t kInputNum = 3; in SigmoidCrossEntropyWithLogitsGradInferShape() local
68 const int64_t kInputNum = 3; in SigmoidCrossEntropyWithLogitsGradInferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/ |
| D | prelu_fp32_coder.cc | 38 constexpr size_t kInputNum = 2; in DoCode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/ |
| D | resize_bilinear_mapper.cc | 25 constexpr auto kInputNum = 2; variable
|
| D | fill_mapper.cc | 27 constexpr auto kInputNum = 3; variable
|