| 189 | bool reshape = false); |
| 190 | |
| 191 | inline Dtype data_at(const int n, const int c, const int h, |
| 192 | const int w) const { |
| 193 | return cpu_data()[offset(n, c, h, w)]; |
| 194 | } |
| 195 | |
| 196 | inline Dtype diff_at(const int n, const int c, const int h, |
| 197 | const int w) const { |
no outgoing calls