Searched refs:kern_depth (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_spatial_convolutions_test.cc | 674 const int kern_depth = 3; in TEST() local
683 Tensor<float, 5> kernel(kern_filters, in_channels, kern_depth, kern_height, in TEST()
697 const int off_p = (kern_depth - 1) / 2; in TEST()
708 for (int p = 0; p < kern_depth; ++p) { in TEST()
735 const int kern_depth = 3; in TEST() local
744 Tensor<float, 5, RowMajor> kernel(kern_width, kern_height, kern_depth, in TEST()
759 const int off_p = (kern_depth - 1) / 2; in TEST()
770 for (int p = 0; p < kern_depth; ++p) { in TEST()
797 const int kern_depth = 3; in TEST() local
806 Tensor<float, 5> kernel(kern_filters, in_channels, kern_depth, kern_height, in TEST()
[all …]
|