| /external/tensorflow/tensorflow/core/kernels/ |
| maxpooling_op.h | 32 typename TTypes<T, 4>::ConstTensor input, int window_rows,
37 Eigen::SpatialMaxPooling(input.swap_layout(), window_cols, window_rows,
45 typename TTypes<qint8, 4>::ConstTensor input, int window_rows,
|
| avgpooling_op.h | 30 typename TTypes<T, 4>::ConstTensor input, int window_rows,
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 | 85 window_rows = GetTensorDim(ksize, data_format, 'H');
97 (depth_window == 1 || (window_rows == 1 && window_cols == 1)),
104 tensor_in_rows, window_rows, row_stride,
179 mkl_params->kernel_size[1] = params.window_rows;
|
| pooling_ops_3d.h | 59 int window_rows; member in struct:tensorflow::Pool3dParameters
|
| avgpooling_op.cc | 167 tensor_in.tensor<T, 4>(), params.window_rows, params.window_cols,
185 typename TTypes<T, 4>::ConstTensor input, int window_rows, \
264 const int window_rows = ksize_[1]; variable
281 GetWindowedOutputSize(in_rows, window_rows, row_stride,
292 in_rows, in_cols, window_rows, window_cols, row_stride,
298 // window_rows could be beyond the input tensor's boundary. In
303 GetBroadcastSize(r, in_rows, window_rows, row_stride,
341 window_rows * window_cols * depth_window * in_rows * in_rows * in_cols;
498 const int window_rows = ksize_[1]; variable
515 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...] |
| pooling_ops_common.cc | 51 window_rows = GetTensorDim(ksize, data_format, 'H');
61 (depth_window == 1 || (window_rows == 1 && window_cols == 1)),
68 context, GetWindowedOutputSize(tensor_in_rows, window_rows, row_stride,
189 .set_window_height(params.window_rows)
332 .set_window_height(params.window_rows)
|
| maxpooling_op.cc | 97 const int32 window_rows = params.window_rows;
124 (hpad < window_rows) ? 0 : (hpad - window_rows) / row_stride + 1;
185 params.depth * params.window_rows *
326 params.out_height, params.out_width, params.window_rows,
574 const int32 window_rows = params.window_rows;
596 const int h_end = std::min(h_start + window_rows, in_rows);
624 params.depth * params.window_rows *
[all...] |
| pooling_ops_3d.cc | 69 window_rows = GetTensorDim(ksize, data_format, '1');
88 GetWindowedOutputSize(tensor_in_rows, window_rows, row_stride,
586 const int32 window_rows = params.window_rows;
614 const int h_end = std::min(h_start + window_rows, in_rows);
649 params.window_planes * params.window_rows * params.window_cols;
[all...] |
| mkl_pooling_ops_common.h | 44 int window_rows; member in struct:tensorflow::MklPoolParameters
68 window_rows(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 | 74 window_rows_(params.window_rows),
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| pooling_ops_test.py | 882 window_rows,
[all...] |
