Home
last modified time | relevance | path

Searched refs:window_cols (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_test.py949 window_cols, argument
1025 window_cols, argument
1104 window_cols=1,
1118 window_cols=2,
1132 window_cols=2,
1146 window_cols=1,
1160 window_cols=2,
1174 window_cols=1,
1188 window_cols=1,
1202 window_cols=2,
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_common.h57 int window_cols; member
168 tensor_in.tensor<T, 4>(), params.window_rows, params.window_cols, in SpatialMaxPool()
203 const int32 window_cols = params.window_cols; in SpatialMaxPool() local
229 const int32 w_start = (wpad < window_cols) in SpatialMaxPool()
231 : (wpad - window_cols) / col_stride + 1; in SpatialMaxPool()
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;
476 const int32 w_start = (wpad < window_cols)
478 : (wpad - window_cols) / col_stride + 1;
[all …]
Dmaxpooling_op.h33 int window_cols, int row_stride, int col_stride, in operator()
37 Eigen::SpatialMaxPooling(input.swap_layout(), window_cols, window_rows, in operator()
46 int window_cols, int row_stride, int col_stride,
Davgpooling_op.h31 int window_cols, int row_stride, int col_stride, in operator()
36 To32Bit(input).swap_layout(), window_cols, window_rows, col_stride, in operator()
41 input.swap_layout(), window_cols, window_rows, col_stride, row_stride, in operator()
Davgpooling_op.cc170 tensor_in.tensor<T, 4>(), params.window_rows, params.window_cols, in Compute()
189 int window_cols, int row_stride, int col_stride, \
272 const int window_cols = ksize_[2]; in Compute() local
291 GetWindowedOutputSize(in_cols, window_cols, col_stride, in Compute()
299 in_rows, in_cols, window_rows, window_cols, row_stride, in Compute()
320 GetBroadcastSize(c, in_cols, window_cols, col_stride, in Compute()
348 window_rows * window_cols * depth_window * in_rows * in_rows * in_cols; in Compute()
512 const int window_cols = ksize_[2]; in Compute() local
531 GetWindowedOutputSize(in_cols, window_cols, col_stride, in Compute()
542 window_cols, // kernel_w in Compute()
Dmkl_pooling_ops_common.cc304 window_cols = GetTensorDim(ksize, data_format, 'W'); in Init()
315 (depth_window == 1 || (window_rows == 1 && window_cols == 1)), in Init()
324 window_cols = GetTensorDim(ksize, data_format, '2'); in Init()
337 (window_rows == 1 && window_cols == 1 && window_planes == 1)), in Init()
356 tensor_in_cols, window_cols, col_stride, in Init()
432 mkl_params->kernel_size[0] = params.window_cols; in ExtractMklOpParams()
Dpooling_ops_common.cc67 window_cols = GetTensorDim(ksize, data_format, 'W'); in PoolParameters()
76 (depth_window == 1 || (window_rows == 1 && window_cols == 1)), in PoolParameters()
86 context, GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride, in PoolParameters()
231 .set_window_width(params.window_cols) in Compute()
379 .set_window_width(params.window_cols) in Compute()
Dmaxpooling_op.cc109 const int32 window_cols = params.window_cols; in SpatialMaxPoolWithArgMaxHelper() local
138 (wpad < window_cols) ? 0 : (wpad - window_cols) / col_stride + 1; in SpatialMaxPoolWithArgMaxHelper()
200 params.window_cols; in SpatialMaxPoolWithArgMaxHelper()
341 params.window_cols, params.row_stride, params.col_stride, params.pad_rows, in MaxPoolingBackwardCustomKernel()
589 const int32 window_cols = params.window_cols; in SpatialMaxPoolGradGrad() local
612 const int w_end = std::min(w_start + window_cols, in_cols); in SpatialMaxPoolGradGrad()
639 params.window_cols; in SpatialMaxPoolGradGrad()
733 params.window_rows, params.window_cols, params.row_stride, in Compute()
1306 params.out_width, params.window_rows, params.window_cols, in launch()
1325 params.out_width, params.window_rows, params.window_cols, in launch()
[all …]
Dpooling_ops_3d.cc70 window_cols = GetTensorDim(ksize, data_format, '2'); in Pool3dParameters()
91 GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride, in Pool3dParameters()
587 const int32 window_cols = params.window_cols; in launch() local
616 const int w_end = std::min(w_start + window_cols, in_cols); in launch()
649 params.window_planes * params.window_rows * params.window_cols; in launch()
813 params.window_planes, params.window_rows, params.window_cols, in launch()
Dpooling_ops_3d.h58 int window_cols; member
Dmkl_pooling_ops_common.h374 int window_cols; member
404 window_cols(0), in MklPoolParameters()
528 memory::dims({pool_params->window_rows, pool_params->window_cols}); in PoolParamsToDims()
539 pool_params->window_cols}); in PoolParamsToDims()
Dpooling_ops_3d_sycl.h75 window_cols_(params.window_cols), in SYCL3DPoolParams()