| /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,
|
| pooling_ops_3d.h | 58 int window_cols; member in struct:tensorflow::Pool3dParameters
|
| avgpooling_op.cc | 170 tensor_in.tensor<T, 4>(), params.window_rows, params.window_cols,
189 int window_cols, int row_stride, int col_stride, \
272 const int window_cols = ksize_[2]; variable
291 GetWindowedOutputSize(in_cols, window_cols, col_stride,
299 in_rows, in_cols, window_rows, window_cols, row_stride,
315 // window_cols could be beyond the input tensor's boundary. In
320 GetBroadcastSize(c, in_cols, window_cols, col_stride,
348 window_rows * window_cols * depth_window * in_rows * in_rows * in_cols;
512 const int window_cols = ksize_[2]; variable
531 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 | 67 window_cols = GetTensorDim(ksize, data_format, 'W');
76 (depth_window == 1 || (window_rows == 1 && window_cols == 1)),
86 context, GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride,
231 .set_window_width(params.window_cols)
379 .set_window_width(params.window_cols)
|
| mkl_pooling_ops_common.cc | 304 window_cols = GetTensorDim(ksize, data_format, 'W');
315 (depth_window == 1 || (window_rows == 1 && window_cols == 1)),
324 window_cols = GetTensorDim(ksize, data_format, '2');
337 (window_rows == 1 && window_cols == 1 && window_planes == 1)),
356 tensor_in_cols, window_cols, col_stride,
432 mkl_params->kernel_size[0] = params.window_cols;
|
| maxpooling_op.cc | 109 const int32 window_cols = params.window_cols;
138 (wpad < window_cols) ? 0 : (wpad - window_cols) / col_stride + 1;
200 params.window_cols;
341 params.window_cols, params.row_stride, params.col_stride, params.pad_rows,
589 const int32 window_cols = params.window_cols;
612 const int w_end = std::min(w_start + window_cols, in_cols);
639 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 | 374 int window_cols; member in struct:tensorflow::MklPoolParameters
404 window_cols(0),
528 memory::dims({pool_params->window_rows, pool_params->window_cols});
539 pool_params->window_cols});
|
| pooling_ops_3d_sycl.h | 75 window_cols_(params.window_cols),
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| pooling_ops_test.py | [all...] |
