| /external/tensorflow/tensorflow/core/kernels/ |
| 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...] |
| 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});
|
