Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h130 const TensorIndex kernelRows = variable
139 kernelRows + (kernelRows - 1) * (row_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,
350 typename internal::traits<Input>::Index kernelRows,
404 kernelRows + (kernelRows - 1) * (row_in_stride - 1);
[all …]
Deigen_backward_cuboid_convolutions.h123 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 …]
Deigen_spatial_convolutions-inl.h1388 const TensorIndex kernelRows =
1394 kernelRows + (kernelRows - 1) * (row_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.h1676 const TensorIndex kernelRows =
1701 out_height = Eigen::divup(inputRows - kernelRows + 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.h127 const TensorIndex kernelRows = variable
133 kernelRows + (kernelRows - 1) * (row_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.h124 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,