| /external/tensorflow/tensorflow/core/kernels/ |
| pooling_ops_3d.h | 59 int window_rows; member in struct:tensorflow::Pool3dParameters
|
| avgpooling_op.cc | 170 tensor_in.tensor<T, 4>(), params.window_rows, params.window_cols,
188 typename TTypes<T, 4>::ConstTensor input, int window_rows, \
271 const int window_rows = ksize_[1]; variable
288 GetWindowedOutputSize(in_rows, window_rows, row_stride,
299 in_rows, in_cols, window_rows, window_cols, row_stride,
305 // window_rows could be beyond the input tensor's boundary. In
310 GetBroadcastSize(r, in_rows, window_rows, row_stride,
348 window_rows * window_cols * depth_window * in_rows * in_rows * in_cols;
511 const int window_rows = ksize_[1]; variable
528 GetWindowedOutputSize(in_rows, window_rows, row_stride
[all...] |
| pooling_ops_common.h | 56 int window_rows; member in struct:tensorflow::PoolParameters
168 tensor_in.tensor<T, 4>(), params.window_rows, params.window_cols,
202 const int32 window_rows = params.window_rows;
225 const int32 h_start = (hpad < window_rows)
227 : (hpad - window_rows) / row_stride + 1;
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;
[all...] |
| mkl_pooling_ops_common.h | 373 int window_rows; member in struct:tensorflow::MklPoolParameters
403 window_rows(0),
528 memory::dims({pool_params->window_rows, pool_params->window_cols});
538 memory::dims({pool_params->window_planes, pool_params->window_rows,
|
