Searched refs:kOutputNum (Results 1 – 14 of 14) sorted by relevance
29 void CustomPredict::set_output_num(const int64_t output_num) { (void)this->AddAttr(kOutputNum, Make… in set_output_num()32 auto value_ptr = this->GetAttr(kOutputNum); in get_output_num()52 shape.push_back(GetValue<int64_t>(primitive->GetAttr(kOutputNum))); in CustomPredictInfer()
33 void Split::set_output_num(const int64_t output_num) { (void)this->AddAttr(kOutputNum, MakeValue(ou… in set_output_num()46 auto value_ptr = GetAttr(kOutputNum); in get_output_num()
146 constexpr auto kOutputNum = "output_num"; variable
22 constexpr int kOutputNum = 1; variable77 if (output_tensors_.size() != kOutputNum) { in CheckInputsOuputs()78 …MS_LOG(ERROR) << "Resize output num should be " << kOutputNum << ", but got " << output_tensors_.s… in CheckInputsOuputs()
24 constexpr size_t kOutputNum = 1; variable31 if (output_tensors_.size() != kOutputNum) { in CheckInputsOutputs()32 …MS_LOG(ERROR) << "Reduce outputs size should be " << kOutputNum << " but got " << output_tensors_.… in CheckInputsOutputs()
31 constexpr int kOutputNum = 1; variable91 if (out_tensors_.size() != kOutputNum) { in CheckInputsOuputs()92 …MS_LOG(ERROR) << "Resize output num should be " << kOutputNum << ", but got " << out_tensors_.size… in CheckInputsOuputs()
32 constexpr size_t kOutputNum = 1; variable40 if (out_tensors_.size() != kOutputNum) { in CheckInputsOutputs()41 …MS_LOG(ERROR) << "Reduce outputs size should be " << kOutputNum << " but got " << out_tensors_.siz… in CheckInputsOutputs()
37 constexpr size_t kOutputNum = 1; variable42 ….size() != kInputNum && in_tensors_.size() != kInputNumOpt) || out_tensors_.size() != kOutputNum) { in Init()44 …<< in_tensors_.size() << ", output size should be" << kOutputNum << ", got " << out_tensors_.size(… in Init()
34 constexpr int kOutputNum = 1; variable46 if (out_tensors_.size() != kOutputNum) { in Init()
36 constexpr size_t kOutputNum = 1; variable53 …e() < kMinInputsSize || in_tensors_.size() > kMaxInputsSize || out_tensors_.size() != kOutputNum) { in Init()55 … << ", got " << in_tensors_.size() << ", output size should be" << kOutputNum << ", got " in Init()
28 constexpr size_t kOutputNum = 1; variable
249 if (inputs.size() < kInputMinNum || outputs.size() != kOutputNum) { in Launch()
29 #define kOutputNum 1 macro
528 const string kOutputNum = "output_num"; in CheckCustomPrimOutputInferResult() local531 auto output_num_value = prim->GetAttr(kOutputNum); in CheckCustomPrimOutputInferResult()