Home
last modified time | relevance | path

Searched refs:kernelFilters (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h119 const TensorIndex kernelFilters = variable
189 post_contract_dims[0] = kernelFilters;
196 post_contract_dims[NumDims - 1] = kernelFilters;
206 kernel_dims[0] = kernelFilters;
210 kernel_dims[1] = kernelFilters;
/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h126 const TensorIndex kernelFilters = variable
215 kernel_dims[0] = kernelFilters * kernelRows * kernelCols;
218 kernel_dims[1] = kernelFilters * kernelRows * kernelCols;
229 pre_contract_dims[0] = kernelFilters * kernelRows * kernelCols;
235 pre_contract_dims[1] = kernelFilters * kernelRows * kernelCols;
445 const TensorIndex kernelFilters = variable
531 pre_contract_dims[1] = kernelRows * kernelCols * kernelFilters;
534 pre_contract_dims[0] = kernelFilters * kernelCols * kernelRows;
548 post_contract_dims[3] = kernelFilters;
550 post_contract_dims[0] = kernelFilters;
Deigen_backward_cuboid_convolutions.h116 const TensorIndex kernelFilters = variable
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;
439 const TensorIndex kernelFilters = variable
527 kernelPlanes * kernelRows * kernelCols * kernelFilters;
531 kernelFilters * kernelCols * kernelRows * kernelPlanes;
546 post_contract_dims[4] = kernelFilters;
548 post_contract_dims[0] = kernelFilters;
Deigen_spatial_convolutions-inl.h1626 const TensorIndex kernelFilters =
1702 post_contract_dims[0] = kernelFilters;
1709 post_contract_dims[NumDims - 1] = kernelFilters;
1719 kernel_dims[0] = kernelFilters;
1723 kernel_dims[1] = kernelFilters;
Deigen_cuboid_convolution.h1862 const TensorIndex kernelFilters =
1916 kernel_dims[0] = kernelFilters;
1920 kernel_dims[1] = kernelFilters;
1956 post_contract_dims[0] = kernelFilters;
1964 post_contract_dims[NumDims - 1] = kernelFilters;