Searched defs:window_rows (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 | 1154 window_rows, argument
1232 window_rows, argument
1534 def _MaxPoolGrad(self, orig_input, orig_output, grad, window_rows, argument
1566 window_rows, window_cols, row_stride, col_stride, argument
2120 def _MaxPoolGradGrad(self, orig_input, orig_output, grad, window_rows, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | pooling_ops_common.h | 64 int window_rows; member
238 const int32_t window_rows = params.window_rows; in SpatialMaxPool() local
|
| H A D | avgpooling_op.cc | 319 const int window_rows = ksize_[1]; in Compute() local
580 const int window_rows = ksize_[1]; in Compute() local
|
| H A D | pooling_ops_3d.h | 59 int window_rows; member
|
| H A D | maxpooling_op.cc | 118 const int32_t window_rows = params.window_rows; in SpatialMaxPoolWithArgMaxHelper() local
620 const int32_t window_rows = params.window_rows; 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 | 605 const int32_t window_rows = params.window_rows; 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 | 375 int window_rows; member
|