Searched defs:window_cols (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | pooling_ops_test.py | 1155 window_cols, argument
1233 window_cols, argument
1535 window_cols, row_stride, col_stride, padding, v2): argument
1566 window_rows, window_cols, row_stride, col_stride, argument
2121 window_cols, row_stride, col_stride, padding): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | pooling_ops_common.h | 65 int window_cols; member
239 const int32_t window_cols = params.window_cols; in SpatialMaxPool() local
|
| H A D | avgpooling_op.cc | 320 const int window_cols = ksize_[2]; in Compute() local
581 const int window_cols = ksize_[2]; in Compute() local
|
| H A D | pooling_ops_3d.h | 58 int window_cols; member
|
| H A D | maxpooling_op.cc | 119 const int32_t window_cols = params.window_cols; in SpatialMaxPoolWithArgMaxHelper() local
621 const int32_t window_cols = params.window_cols; in SpatialMaxPoolGradGrad() local
|
| H A D | pooling_ops_common.cc | 87 Status CheckPaddingSize(int64_t window_rows, int64_t window_cols, in CheckPaddingSize()
|
| H A D | pooling_ops_3d.cc | 606 const int32_t window_cols = params.window_cols; in launch() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/pooling/ |
| H A D | pooling_depthfirst.hpp | 44 DepthfirstStrategy(unsigned int window_rows, unsigned int window_cols, in DepthfirstStrategy()
|
| H A D | pooling_depthfirst_generic.hpp | 118 const unsigned int window_rows, window_cols; member in arm_conv::pooling::GenericDepthfirstWrapper
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_pooling_ops_common.h | 376 int window_cols; member
|