/external/tensorflow/tensorflow/python/kernel_tests/ |
D | pooling_ops_test.py | 949 window_cols, argument 1025 window_cols, argument 1104 window_cols=1, 1118 window_cols=2, 1132 window_cols=2, 1146 window_cols=1, 1160 window_cols=2, 1174 window_cols=1, 1188 window_cols=1, 1202 window_cols=2, [all …]
|
/external/tensorflow/tensorflow/core/kernels/ |
D | pooling_ops_common.h | 57 int window_cols; member 168 tensor_in.tensor<T, 4>(), params.window_rows, params.window_cols, in SpatialMaxPool() 203 const int32 window_cols = params.window_cols; in SpatialMaxPool() local 229 const int32 w_start = (wpad < window_cols) in SpatialMaxPool() 231 : (wpad - window_cols) / col_stride + 1; in SpatialMaxPool() 276 params.window_cols, params.row_stride, params.col_stride, 412 tensor_in.tensor<T, 4>(), params.window_rows, params.window_cols, 450 const int32 window_cols = params.window_cols; 476 const int32 w_start = (wpad < window_cols) 478 : (wpad - window_cols) / col_stride + 1; [all …]
|
D | maxpooling_op.h | 33 int window_cols, int row_stride, int col_stride, in operator() 37 Eigen::SpatialMaxPooling(input.swap_layout(), window_cols, window_rows, in operator() 46 int window_cols, int row_stride, int col_stride,
|
D | avgpooling_op.h | 31 int window_cols, int row_stride, int col_stride, in operator() 36 To32Bit(input).swap_layout(), window_cols, window_rows, col_stride, in operator() 41 input.swap_layout(), window_cols, window_rows, col_stride, row_stride, in operator()
|
D | avgpooling_op.cc | 170 tensor_in.tensor<T, 4>(), params.window_rows, params.window_cols, in Compute() 189 int window_cols, int row_stride, int col_stride, \ 272 const int window_cols = ksize_[2]; in Compute() local 291 GetWindowedOutputSize(in_cols, window_cols, col_stride, in Compute() 299 in_rows, in_cols, window_rows, window_cols, row_stride, in Compute() 320 GetBroadcastSize(c, in_cols, window_cols, col_stride, in Compute() 348 window_rows * window_cols * depth_window * in_rows * in_rows * in_cols; in Compute() 512 const int window_cols = ksize_[2]; in Compute() local 531 GetWindowedOutputSize(in_cols, window_cols, col_stride, in Compute() 542 window_cols, // kernel_w in Compute()
|
D | mkl_pooling_ops_common.cc | 304 window_cols = GetTensorDim(ksize, data_format, 'W'); in Init() 315 (depth_window == 1 || (window_rows == 1 && window_cols == 1)), in Init() 324 window_cols = GetTensorDim(ksize, data_format, '2'); in Init() 337 (window_rows == 1 && window_cols == 1 && window_planes == 1)), in Init() 356 tensor_in_cols, window_cols, col_stride, in Init() 432 mkl_params->kernel_size[0] = params.window_cols; in ExtractMklOpParams()
|
D | pooling_ops_common.cc | 67 window_cols = GetTensorDim(ksize, data_format, 'W'); in PoolParameters() 76 (depth_window == 1 || (window_rows == 1 && window_cols == 1)), in PoolParameters() 86 context, GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride, in PoolParameters() 231 .set_window_width(params.window_cols) in Compute() 379 .set_window_width(params.window_cols) in Compute()
|
D | maxpooling_op.cc | 109 const int32 window_cols = params.window_cols; in SpatialMaxPoolWithArgMaxHelper() local 138 (wpad < window_cols) ? 0 : (wpad - window_cols) / col_stride + 1; in SpatialMaxPoolWithArgMaxHelper() 200 params.window_cols; in SpatialMaxPoolWithArgMaxHelper() 341 params.window_cols, params.row_stride, params.col_stride, params.pad_rows, in MaxPoolingBackwardCustomKernel() 589 const int32 window_cols = params.window_cols; in SpatialMaxPoolGradGrad() local 612 const int w_end = std::min(w_start + window_cols, in_cols); in SpatialMaxPoolGradGrad() 639 params.window_cols; in SpatialMaxPoolGradGrad() 733 params.window_rows, params.window_cols, params.row_stride, in Compute() 1306 params.out_width, params.window_rows, params.window_cols, in launch() 1325 params.out_width, params.window_rows, params.window_cols, in launch() [all …]
|
D | pooling_ops_3d.cc | 70 window_cols = GetTensorDim(ksize, data_format, '2'); in Pool3dParameters() 91 GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride, in Pool3dParameters() 587 const int32 window_cols = params.window_cols; in launch() local 616 const int w_end = std::min(w_start + window_cols, in_cols); in launch() 649 params.window_planes * params.window_rows * params.window_cols; in launch() 813 params.window_planes, params.window_rows, params.window_cols, in launch()
|
D | pooling_ops_3d.h | 58 int window_cols; member
|
D | mkl_pooling_ops_common.h | 374 int window_cols; member 404 window_cols(0), in MklPoolParameters() 528 memory::dims({pool_params->window_rows, pool_params->window_cols}); in PoolParamsToDims() 539 pool_params->window_cols}); in PoolParamsToDims()
|
D | pooling_ops_3d_sycl.h | 75 window_cols_(params.window_cols), in SYCL3DPoolParams()
|