Searched refs:kernelRows (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | eigen_spatial_convolutions.h | 124 const TensorIndex kernelRows = variable
130 kernelRows + (kernelRows - 1) * (row_in_stride - 1);
168 pre_contract_dims[0] = kernelChannels * kernelRows * kernelCols;
174 pre_contract_dims[1] = kernelChannels * kernelRows * kernelCols;
207 kernel_dims[1] = kernelChannels * kernelRows * kernelCols;
209 kernel_dims[0] = kernelChannels * kernelRows * kernelCols;
215 .extract_image_patches(kernelRows, kernelCols, row_stride, col_stride,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions.h | 131 const TensorIndex kernelRows = variable
140 kernelRows + (kernelRows - 1) * (row_in_stride - 1);
215 kernel_dims[0] = kernelFilters * kernelRows * kernelCols;
218 kernel_dims[1] = kernelFilters * kernelRows * kernelCols;
229 pre_contract_dims[0] = kernelFilters * kernelRows * kernelCols;
235 pre_contract_dims[1] = kernelFilters * kernelRows * kernelCols;
286 kernelRows, kernelCols, 1, 1, row_in_stride,
293 .extract_image_patches(kernelRows, kernelCols, 1, 1, row_in_stride,
401 typename internal::traits<Input>::Index kernelRows,
456 kernelRows + (kernelRows - 1) * (row_in_stride - 1);
[all …]
|
| D | eigen_backward_cuboid_convolutions.h | 123 const TensorIndex kernelRows = variable
139 const TensorIndex kernelRowsEff = kernelRows;
212 kernel_dims[0] = kernelFilters * kernelPlanes * kernelRows * kernelCols;
215 kernel_dims[1] = kernelFilters * kernelPlanes * kernelRows * kernelCols;
228 kernelFilters * kernelPlanes * kernelRows * kernelCols;
235 kernelFilters * kernelPlanes * kernelRows * kernelCols;
282 kernelPlanes, kernelRows, kernelCols, 1, 1, 1,
290 .extract_volume_patches(kernelPlanes, kernelRows, kernelCols, 1, 1, 1,
392 typename internal::traits<Input>::Index kernelRows,
453 const TensorIndex kernelRowsEff = kernelRows;
[all …]
|
| D | eigen_spatial_convolutions-inl.h | 1631 const TensorIndex kernelRows =
1637 kernelRows + (kernelRows - 1) * (row_in_stride - 1);
1681 pre_contract_dims[0] = kernelChannels * kernelRows * kernelCols;
1687 pre_contract_dims[1] = kernelChannels * kernelRows * kernelCols;
1720 kernel_dims[1] = kernelChannels * kernelRows * kernelCols;
1722 kernel_dims[0] = kernelChannels * kernelRows * kernelCols;
1731 kernelRows, kernelCols, row_stride, col_stride,
1741 .extract_image_patches(kernelRows, kernelCols, row_stride,
1756 kernelRows, kernelCols, row_stride, col_stride,
1762 .extract_image_patches(kernelRows, kernelCols, row_stride,
|
| D | eigen_cuboid_convolution.h | 1870 const TensorIndex kernelRows =
1895 out_height = Eigen::divup(inputRows - kernelRows + 1,
1917 kernel_dims[1] = kernelChannels * kernelPlanes * kernelRows * kernelCols;
1919 kernel_dims[0] = kernelChannels * kernelPlanes * kernelRows * kernelCols;
1930 kernelChannels * kernelPlanes * kernelRows * kernelCols;
1937 kernelChannels * kernelPlanes * kernelRows * kernelCols;
1978 kernelPlanes, kernelRows, kernelCols, stridePlanes,
1984 .extract_volume_patches(kernelPlanes, kernelRows, kernelCols,
|