| /external/tensorflow/tensorflow/core/kernels/ |
| maxpooling_op.h | 33 int window_cols, int row_stride, int col_stride,
37 Eigen::SpatialMaxPooling(input.swap_layout(), window_cols, window_rows,
46 int window_cols, int row_stride, int col_stride,
|
| avgpooling_op.h | 31 int window_cols, int row_stride, int col_stride,
36 To32Bit(input).swap_layout(), window_cols, window_rows, col_stride,
41 input.swap_layout(), window_cols, window_rows, col_stride, row_stride,
|
| mkl_pooling_ops_common.cc | 86 window_cols = GetTensorDim(ksize, data_format, 'W');
97 (depth_window == 1 || (window_rows == 1 && window_cols == 1)),
108 tensor_in_cols, window_cols, col_stride,
178 mkl_params->kernel_size[0] = params.window_cols;
|
| pooling_ops_3d.h | 58 int window_cols; member in struct:tensorflow::Pool3dParameters
|
| avgpooling_op.cc | 167 tensor_in.tensor<T, 4>(), params.window_rows, params.window_cols,
186 int window_cols, int row_stride, int col_stride, \
265 const int window_cols = ksize_[2]; variable
284 GetWindowedOutputSize(in_cols, window_cols, col_stride,
292 in_rows, in_cols, window_rows, window_cols, row_stride,
308 // window_cols could be beyond the input tensor's boundary. In
313 GetBroadcastSize(c, in_cols, window_cols, col_stride,
341 window_rows * window_cols * depth_window * in_rows * in_rows * in_cols;
499 const int window_cols = ksize_[2]; variable
518 GetWindowedOutputSize(in_cols, window_cols, col_stride
[all...] |
| pooling_ops_common.h | 57 int window_cols; member in struct:tensorflow::PoolParameters
168 tensor_in.tensor<T, 4>(), params.window_rows, params.window_cols,
203 const int32 window_cols = params.window_cols;
229 const int32 w_start = (wpad < window_cols)
231 : (wpad - window_cols) / col_stride + 1;
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;
[all...] |
| pooling_ops_common.cc | 52 window_cols = GetTensorDim(ksize, data_format, 'W');
61 (depth_window == 1 || (window_rows == 1 && window_cols == 1)),
71 context, GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride,
190 .set_window_width(params.window_cols)
333 .set_window_width(params.window_cols)
|
| maxpooling_op.cc | 98 const int32 window_cols = params.window_cols;
127 (wpad < window_cols) ? 0 : (wpad - window_cols) / col_stride + 1;
186 params.window_cols;
327 params.window_cols, params.row_stride, params.col_stride, params.pad_rows,
575 const int32 window_cols = params.window_cols;
598 const int w_end = std::min(w_start + window_cols, in_cols);
625 params.window_cols;
[all...] |
| pooling_ops_3d.cc | 70 window_cols = GetTensorDim(ksize, data_format, '2');
91 GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride,
587 const int32 window_cols = params.window_cols;
616 const int w_end = std::min(w_start + window_cols, in_cols);
649 params.window_planes * params.window_rows * params.window_cols;
[all...] |
| mkl_pooling_ops_common.h | 45 int window_cols; member in struct:tensorflow::MklPoolParameters
69 window_cols(0),
|
| mkl_avgpooling_op.cc | 511 memory::dims({pool_params.window_rows, pool_params.window_cols}),
605 memory::dims({pool_params.window_rows, pool_params.window_cols}),
618 memory::dims({pool_params.window_rows, pool_params.window_cols}),
|
| mkl_maxpooling_op.cc | 555 memory::dims({pool_params.window_rows, pool_params.window_cols}),
668 memory::dims({pool_params.window_rows, pool_params.window_cols}),
681 memory::dims({pool_params.window_rows, pool_params.window_cols}),
[all...] |
| pooling_ops_3d_sycl.h | 75 window_cols_(params.window_cols),
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| pooling_ops_test.py | 883 window_cols,
[all...] |
