HomeSort by relevance Sort by last modified time
    Searched defs:kernelRows (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
eigen_spatial_convolutions.h 127 const TensorIndex kernelRows =
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/core/kernels/
eigen_cuboid_convolution.h 103 const TensorIndex kernelRows =
128 out_height = Eigen::divup(inputRows - kernelRows + 1,
150 kernel_dims[1] = kernelChannels * kernelDepth * kernelRows * kernelCols;
152 kernel_dims[0] = kernelChannels * kernelDepth * kernelRows * kernelCols;
163 kernelChannels * kernelDepth * kernelRows * kernelCols;
170 kernelChannels * kernelDepth * kernelRows * kernelCols;
211 kernelDepth, kernelRows, kernelCols, stridePlanes,
217 .extract_volume_patches(kernelDepth, kernelRows, kernelCols,
eigen_backward_spatial_convolutions.h 130 const TensorIndex kernelRows =
139 kernelRows + (kernelRows - 1) * (row_in_stride - 1);
209 kernel_dims[0] = kernelFilters * kernelRows * kernelCols;
212 kernel_dims[1] = kernelFilters * kernelRows * kernelCols;
223 pre_contract_dims[0] = kernelFilters * kernelRows * kernelCols;
229 pre_contract_dims[1] = kernelFilters * kernelRows * kernelCols;
237 // the kernelRows and the kernelCols.
275 kernelRows, kernelCols, 1, 1, row_in_stride,
282 .extract_image_patches(kernelRows, kernelCols, 1, 1, row_in_stride
    [all...]
eigen_backward_cuboid_convolutions.h 116 const TensorIndex kernelRows =
142 0, (size_y - 1) * strideRows + kernelRows - inputRows);
156 const TensorIndex padding_top = kernelRows - 1 - forward_pad_y;
162 const TensorIndex padding_bottom = inputRows + kernelRows - 1 -
198 kernel_dims[2] = kernelRows * kernelCols * kernelPlanes;
200 kernel_dims[0] = kernelRows * kernelCols * kernelPlanes;
214 pre_contract_dims[1] = kernelRows * kernelCols * kernelPlanes;
221 pre_contract_dims[1] = kernelRows * kernelCols * kernelPlanes;
284 kernelPlanes, kernelRows, kernelCols, 1, 1, 1,
292 .extract_volume_patches(kernelPlanes, kernelRows, kernelCols, 1, 1, 1
    [all...]

Completed in 749 milliseconds