Commit 6a073b39 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

First pass at enabling non-square strides for convolutions: 

- Change the stride, and in_stride template arguments of Eigen
  SpatialConvolution, SpatialConvolutionBackwardInput, and
  SpatialConvolutionBackwardKernel to row_stride, col_stride, row_in_stride,
  col_in_stride.

- Change tensorflow kernels to pass the additional stride parameters.

- Rationalize the place where we swap row/col: swap just before calling Eigen.

This just enables the plumbing. Non-square strides are still forbidden in the ops.
Change: 118484322
parent 622daf83
Expand all Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment