Searched refs:kInputSize (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | soft_margin_loss.cc | 26 constexpr int64_t kInputSize = 2; variable
31 …rtUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kEqual, kInputSize, op_name); in SoftMarginLossInferShape()
46 …rtUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kEqual, kInputSize, op_name); in SoftMarginLossInferType()
|
| D | ctc_loss_v2_grad.cc | 30 constexpr int64_t kInputSize = 7; variable
36 …ckAndConvertUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kEqual, kInputSize, in CTCLossV2GradInferShape()
|
| D | ctc_loss_v2.cc | 33 constexpr int64_t kInputSize = 4; variable
38 …ckAndConvertUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kEqual, kInputSize, in CTCLossV2InferShape()
|
| D | detection_post_process.cc | 42 (void)this->AddAttr(kInputSize, MakeValue(inputSize)); in set_input_size()
46 auto value_ptr = this->GetAttr(kInputSize); in get_input_size()
|
| D | op_utils.h | 98 constexpr auto kInputSize = "input_size"; variable
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | soft_margin_loss_grad.cc | 26 constexpr int64_t kInputSize = 3; variable
31 …rtUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kEqual, kInputSize, op_name); in SoftMarginLossGradInferShape()
44 …rtUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kEqual, kInputSize, op_name); in SoftMarginLossGradInferType()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | check_valid_cpu_kernel.cc | 25 constexpr size_t kInputSize = 2; variable
78 if (inputs.size() != kInputSize) { in CheckParams()
79 …) << "Input number is: " << inputs.size() << ", but CheckValid needs " << kInputSize << " inputs."; in CheckParams()
|
| D | roi_align_cpu_kernel.cc | 24 constexpr size_t kInputSize = 2; variable
121 if (inputs.size() != kInputSize) { in CheckParam()
122 …ON) << "Input number is: " << inputs.size() << ", but ROIAlign needs " << kInputSize << " inputs."; in CheckParam()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tflite_rel_pos_multi_head_attention_fusion.cc | 33 const size_t kInputSize = 16; variable
290 auto quant_params_holder = std::make_shared<lite::QuantParamHolder>(kInputSize, kOutputSize); in SetQuantParamForAttentionNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | helper.cc | 370 constexpr auto kInputSize = 2; in RemoveNopNode() local
371 if (cinput->inputs().size() == kInputSize) { in RemoveNopNode()
|