| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | pooling_ops_test.py | 948 window_rows, argument
993 ksize = [1, 1, window_rows, window_rows]
997 ksize = [1, window_rows, window_rows, 1]
1024 window_rows, argument
1069 ksize = [1, 1, window_rows, window_rows]
1073 ksize = [1, window_rows, window_rows, 1]
1103 window_rows=1,
1117 window_rows=2,
1131 window_rows=2,
1145 window_rows=1,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common.h | 56 int window_rows; member
168 tensor_in.tensor<T, 4>(), params.window_rows, params.window_cols, in SpatialMaxPool()
202 const int32 window_rows = params.window_rows; in SpatialMaxPool() local
225 const int32 h_start = (hpad < window_rows) in SpatialMaxPool()
227 : (hpad - window_rows) / row_stride + 1; in SpatialMaxPool()
275 params.depth, params.out_height, params.out_width, params.window_rows,
412 tensor_in.tensor<T, 4>(), params.window_rows, params.window_cols,
449 const int32 window_rows = params.window_rows;
472 const int32 h_start = (hpad < window_rows)
474 : (hpad - window_rows) / row_stride + 1;
[all …]
|
| D | maxpooling_op.h | 32 typename TTypes<T, 4>::ConstTensor input, int window_rows, in operator()
37 Eigen::SpatialMaxPooling(input.swap_layout(), window_cols, window_rows, in operator()
45 typename TTypes<qint8, 4>::ConstTensor input, int window_rows,
|
| D | avgpooling_op.h | 30 typename TTypes<T, 4>::ConstTensor input, int window_rows, 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()
188 typename TTypes<T, 4>::ConstTensor input, int window_rows, \
271 const int window_rows = ksize_[1]; in Compute() local
288 GetWindowedOutputSize(in_rows, window_rows, row_stride, in Compute()
299 in_rows, in_cols, window_rows, window_cols, row_stride, in Compute()
310 GetBroadcastSize(r, in_rows, window_rows, row_stride, in Compute()
348 window_rows * window_cols * depth_window * in_rows * in_rows * in_cols; in Compute()
511 const int window_rows = ksize_[1]; in Compute() local
528 GetWindowedOutputSize(in_rows, window_rows, row_stride, in Compute()
541 window_rows, // kernel_h in Compute()
|
| D | mkl_pooling_ops_common.cc | 303 window_rows = GetTensorDim(ksize, data_format, 'H'); in Init()
315 (depth_window == 1 || (window_rows == 1 && window_cols == 1)), in Init()
323 window_rows = GetTensorDim(ksize, data_format, '1'); in Init()
337 (window_rows == 1 && window_cols == 1 && window_planes == 1)), in Init()
352 tensor_in_rows, window_rows, row_stride, in Init()
433 mkl_params->kernel_size[1] = params.window_rows; in ExtractMklOpParams()
|
| D | pooling_ops_common.cc | 66 window_rows = GetTensorDim(ksize, data_format, 'H'); in PoolParameters()
76 (depth_window == 1 || (window_rows == 1 && window_cols == 1)), in PoolParameters()
83 context, GetWindowedOutputSize(tensor_in_rows, window_rows, row_stride, in PoolParameters()
230 .set_window_height(params.window_rows) in Compute()
378 .set_window_height(params.window_rows) in Compute()
|
| D | maxpooling_op.cc | 108 const int32 window_rows = params.window_rows; in SpatialMaxPoolWithArgMaxHelper() local
135 (hpad < window_rows) ? 0 : (hpad - window_rows) / row_stride + 1; in SpatialMaxPoolWithArgMaxHelper()
199 params.depth * params.window_rows * in SpatialMaxPoolWithArgMaxHelper()
340 params.out_height, params.out_width, params.window_rows, in MaxPoolingBackwardCustomKernel()
588 const int32 window_rows = params.window_rows; in SpatialMaxPoolGradGrad() local
610 const int h_end = std::min(h_start + window_rows, in_rows); in SpatialMaxPoolGradGrad()
638 params.depth * params.window_rows * 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 | 69 window_rows = GetTensorDim(ksize, data_format, '1'); in Pool3dParameters()
88 GetWindowedOutputSize(tensor_in_rows, window_rows, row_stride, in Pool3dParameters()
586 const int32 window_rows = params.window_rows; in launch() local
614 const int h_end = std::min(h_start + window_rows, in_rows); 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 | 59 int window_rows; member
|
| D | mkl_pooling_ops_common.h | 373 int window_rows; member
403 window_rows(0), in MklPoolParameters()
528 memory::dims({pool_params->window_rows, pool_params->window_cols}); in PoolParamsToDims()
538 memory::dims({pool_params->window_planes, pool_params->window_rows, in PoolParamsToDims()
|
| D | pooling_ops_3d_sycl.h | 74 window_rows_(params.window_rows), in SYCL3DPoolParams()
|