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

  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
eigen_spatial_convolutions.h 129 const TensorIndex kernelCols =
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/core/kernels/
eigen_cuboid_convolution.h 105 const TensorIndex kernelCols =
130 out_width = Eigen::divup(inputCols - kernelCols + 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 132 const TensorIndex kernelCols =
141 kernelCols + (kernelCols - 1) * (col_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 118 const TensorIndex kernelCols =
144 0, (size_x - 1) * strideCols + kernelCols - inputCols);
157 const TensorIndex padding_left = kernelCols - 1 - forward_pad_x;
165 const TensorIndex padding_right = inputCols + kernelCols - 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 104 milliseconds