Searched refs:kernelPlanes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_cuboid_convolutions.h | 121 const TensorIndex kernelPlanes = variable
138 const TensorIndex kernelPlanesEff = kernelPlanes;
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,
391 typename internal::traits<Input>::Index kernelPlanes,
452 const TensorIndex kernelPlanesEff = kernelPlanes;
[all …]
|
| D | eigen_cuboid_convolution.h | 1674 const TensorIndex kernelPlanes =
1699 out_planes = Eigen::divup(inputPlanes - kernelPlanes + 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,
|