Searched refs:kernelCols (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | eigen_spatial_convolutions.h | 126 const TensorIndex kernelCols = variable
132 kernelCols + (kernelCols - 1) * (col_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 | 133 const TensorIndex kernelCols = variable
142 kernelCols + (kernelCols - 1) * (col_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,
402 typename internal::traits<Input>::Index kernelCols,
458 kernelCols + (kernelCols - 1) * (col_in_stride - 1);
[all …]
|
| D | eigen_backward_cuboid_convolutions.h | 125 const TensorIndex kernelCols = variable
140 const TensorIndex kernelColsEff = kernelCols;
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,
393 typename internal::traits<Input>::Index kernelCols,
454 const TensorIndex kernelColsEff = kernelCols;
[all …]
|
| D | eigen_spatial_convolutions-inl.h | 1633 const TensorIndex kernelCols =
1639 kernelCols + (kernelCols - 1) * (col_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 | 1872 const TensorIndex kernelCols =
1897 out_width = Eigen::divup(inputCols - kernelCols + 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,
|