Searched defs:set_output_num (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ | ||
| D | split.cc | 33 void Split::set_output_num(const int64_t output_num) { (void)this->AddAttr(kOutputNum, MakeValue(ou… in set_output_num() function in mindspore::ops::Split |
| D | custom_predict.cc | 29 void CustomPredict::set_output_num(const int64_t output_num) { (void)this->AddAttr(kOutputNum, Make… in set_output_num() function in mindspore::ops::CustomPredict |