Searched refs:strideRows (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | eigen_pooling.h | 65 DenseIndex patchCols, DenseIndex strideRows, 98 strideRows); 104 static_cast<DenseIndex>(in.dimension(idxRows)), strideRows); 132 patchRows, patchCols, strideRows, strideCols, in_strideRows, 182 DenseIndex stridePlanes, DenseIndex strideRows, in CuboidMaxPooling() argument 214 strideRows); in CuboidMaxPooling() 222 static_cast<DenseIndex>(in.dimension(idxRows)), strideRows); in CuboidMaxPooling() 251 strideRows, strideCols, padding_type, in CuboidMaxPooling() 422 DenseIndex patchCols, DenseIndex strideRows, 455 strideRows); [all …]
|
D | eigen_backward_cuboid_convolutions.h | 394 const DenseIndex stridePlanes = 1, const DenseIndex strideRows = 1, 460 0, (outputRows - 1) * strideRows + kernelRowsEff - inputRows); 470 const auto expanded_out_rows = (outputRows - 1) * strideRows + 1; 590 stridePlanes, strideRows, strideCols, 598 stridePlanes, strideRows, strideCols, top_pad_planes,
|
D | eigen_cuboid_convolution.h | 1647 const Index stridePlanes = 1, const Index strideRows = 1, 1702 static_cast<TensorIndex>(strideRows)); 1710 Eigen::divup(inputRows, static_cast<TensorIndex>(strideRows)); 1785 strideRows, strideCols, padding_type) 1791 stridePlanes, strideRows, strideCols,
|