* @brief Sets the loss associated with a top blob at a given index. */
| 204 | * @brief Sets the loss associated with a top blob at a given index. |
| 205 | */ |
| 206 | inline void set_loss(const int top_index, const Dtype value) { |
| 207 | if (loss_.size() <= top_index) { |
| 208 | loss_.resize(top_index + 1, Dtype(0)); |
| 209 | } |
| 210 | loss_[top_index] = value; |
| 211 | } |
| 212 | |
| 213 | /** |
| 214 | * @brief Returns the layer type. |
no test coverage detected