diff options
Diffstat (limited to 'tensorflow/core/kernels/conv_grad_ops.h')
| -rw-r--r-- | tensorflow/core/kernels/conv_grad_ops.h | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/conv_grad_ops.h b/tensorflow/core/kernels/conv_grad_ops.h index 535586d53a..9551959463 100644 --- a/tensorflow/core/kernels/conv_grad_ops.h +++ b/tensorflow/core/kernels/conv_grad_ops.h @@ -234,6 +234,16 @@ struct ConvBackpropDimensions { // Input and output feature depth. int64 in_depth, out_depth; + + // Convenience access methods for spatial dimensions properties. + int64 input_size(int dim) const { return spatial_dims[dim].input_size; } + int64 filter_size(int dim) const { return spatial_dims[dim].filter_size; } + int64 output_size(int dim) const { return spatial_dims[dim].output_size; } + int64 stride(int dim) const { return spatial_dims[dim].stride; } + int64 dilation(int dim) const { return spatial_dims[dim].dilation; } + + // Compute padding for the given spatial dimension. + int SpatialPadding(const Padding& padding, int dim) const; }; // Common code between implementations of Conv?DBackpropInput and |