| 30 | Filter3DDesc filter_desc; |
| 31 | Conv3DDesc conv_desc; |
| 32 | void set( |
| 33 | const TensorLayout& src, const CanonizedFilterMeta& filter, |
| 34 | const TensorLayout& dst, const param::Convolution3D& param) { |
| 35 | src_desc.set(src); |
| 36 | filter_desc.set(filter); |
| 37 | dst_desc.set(dst); |
| 38 | conv_desc.set(param, filter.group); |
| 39 | } |
| 40 | }; |
| 41 | |
| 42 | struct CUDNNBwdDataDescs { |