| 525 | // -- ConvolutionDescriptor |
| 526 | |
| 527 | ConvolutionDescriptor::ConvolutionDescriptor(int ndims) { |
| 528 | proto_.mutable_paddings()->Resize(ndims, 0); |
| 529 | proto_.mutable_strides()->Resize(ndims, 1); |
| 530 | proto_.mutable_dilations()->Resize(ndims, 1); |
| 531 | proto_.set_group_count(1); |
| 532 | proto_.set_convolution_mode(ConvolutionMode::CROSS_CORRELATION); |
| 533 | } |
| 534 | |
| 535 | ConvolutionDescriptor::ConvolutionDescriptor() |
| 536 | : ConvolutionDescriptor(/*ndims=*/2) {} |