Searched refs:strideCols (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_pooling.h | 66 DenseIndex strideCols, const PaddingType padding_type,
101 strideCols);
106 static_cast<DenseIndex>(in.dimension(idxCols)), strideCols);
132 patchRows, patchCols, strideRows, strideCols, in_strideRows,
183 DenseIndex strideCols, const PaddingType padding_type) { in CuboidMaxPooling() argument
217 strideCols); in CuboidMaxPooling()
224 static_cast<DenseIndex>(in.dimension(idxCols)), strideCols); in CuboidMaxPooling()
251 strideRows, strideCols, padding_type, in CuboidMaxPooling()
423 DenseIndex strideCols, const PaddingType padding_type,
458 strideCols);
[all …]
|
| D | eigen_backward_cuboid_convolutions.h | 395 const DenseIndex strideCols = 1) {
462 0, (outputCols - 1) * strideCols + kernelColsEff - inputCols);
471 const auto expanded_out_cols = (outputCols - 1) * strideCols + 1;
590 stridePlanes, strideRows, strideCols,
598 stridePlanes, strideRows, strideCols, top_pad_planes,
|
| D | eigen_cuboid_convolution.h | 1648 const Index strideCols = 1,
1704 static_cast<TensorIndex>(strideCols));
1711 out_width = Eigen::divup(inputCols, static_cast<TensorIndex>(strideCols));
1785 strideRows, strideCols, padding_type)
1791 stridePlanes, strideRows, strideCols,
|