| 72 | // ----------------------------------------------------------------------------------------------- |
| 73 | |
| 74 | void ConcatConv::setWeight(const Ref<Tensor>& weight) |
| 75 | { |
| 76 | if (!weight || weight->getDesc() != weightDesc) |
| 77 | throw std::invalid_argument("invalid concat+conv weight"); |
| 78 | |
| 79 | this->weight = weight; |
| 80 | updateWeight(); |
| 81 | } |
| 82 | |
| 83 | // ----------------------------------------------------------------------------------------------- |
| 84 |