Home
last modified time | relevance | path

Searched refs:kernelCols (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h132 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 …]
Deigen_backward_cuboid_convolutions.h125 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 …]
Deigen_spatial_convolutions-inl.h1390 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,
Deigen_cuboid_convolution.h1678 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/
Deigen_spatial_convolutions.h129 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/
Deigen_spatial_convolutions.h126 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,