Searched refs:kernelCols (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | eigen_backward_spatial_convolutions.h | 132 const TensorIndex kernelCols = variable 141 kernelCols + (kernelCols - 1) * (col_in_stride - 1); 214 kernel_dims[0] = kernelFilters * kernelRows * kernelCols; 217 kernel_dims[1] = kernelFilters * kernelRows * kernelCols; 228 pre_contract_dims[0] = kernelFilters * kernelRows * kernelCols; 234 pre_contract_dims[1] = kernelFilters * kernelRows * kernelCols; 280 kernelRows, kernelCols, 1, 1, row_in_stride, 287 .extract_image_patches(kernelRows, kernelCols, 1, 1, row_in_stride, 351 typename internal::traits<Input>::Index kernelCols, 406 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 | 1390 const TensorIndex kernelCols = 1396 kernelCols + (kernelCols - 1) * (col_in_stride - 1); 1432 pre_contract_dims[0] = kernelChannels * kernelRows * kernelCols; 1438 pre_contract_dims[1] = kernelChannels * kernelRows * kernelCols; 1471 kernel_dims[1] = kernelChannels * kernelRows * kernelCols; 1473 kernel_dims[0] = kernelChannels * kernelRows * kernelCols; 1481 kernelRows, kernelCols, row_stride, col_stride, 1487 .extract_image_patches(kernelRows, kernelCols, row_stride, col_stride,
|
D | eigen_cuboid_convolution.h | 1678 const TensorIndex kernelCols = 1703 out_width = Eigen::divup(inputCols - kernelCols + 1, 1723 kernel_dims[1] = kernelChannels * kernelPlanes * kernelRows * kernelCols; 1725 kernel_dims[0] = kernelChannels * kernelPlanes * kernelRows * kernelCols; 1736 kernelChannels * kernelPlanes * kernelRows * kernelCols; 1743 kernelChannels * kernelPlanes * kernelRows * kernelCols; 1784 kernelPlanes, kernelRows, kernelCols, stridePlanes, 1790 .extract_volume_patches(kernelPlanes, kernelRows, kernelCols,
|
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
D | eigen_spatial_convolutions.h | 129 const TensorIndex kernelCols = variable 135 kernelCols + (kernelCols - 1) * (col_in_stride - 1); 171 pre_contract_dims[0] = kernelChannels * kernelRows * kernelCols; 177 pre_contract_dims[1] = kernelChannels * kernelRows * kernelCols; 210 kernel_dims[1] = kernelChannels * kernelRows * kernelCols; 212 kernel_dims[0] = kernelChannels * kernelRows * kernelCols; 218 .extract_image_patches(kernelRows, kernelCols, row_stride, col_stride,
|
/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,
|