| 59 | FilterDesc<param::Convolution> grad_desc; |
| 60 | ConvDesc conv_desc; |
| 61 | void set( |
| 62 | const TensorLayout& src, const TensorLayout& diff, |
| 63 | const CanonizedFilterMeta& grad, const param::Convolution& param) { |
| 64 | src_desc.set(src, param.format); |
| 65 | diff_desc.set(diff, param.format); |
| 66 | grad_desc.set(grad); |
| 67 | conv_desc.set(src.dtype, param, grad.group); |
| 68 | } |
| 69 | }; |
| 70 | |
| 71 | /*! |