Home | History | Annotate | Download | only in kernels

Lines Matching defs:output_cols

39   const int output_cols = input_cols - patch_cols + 1;
43 Tensor<float, 3> output_backward(output_depth, output_rows, output_cols);
66 output_j < output_cols) {
88 const int output_cols = input_cols - patch_cols + 1;
94 Tensor<float, 3, RowMajor> output_backward(output_cols, output_rows,
118 output_j < output_cols) {
141 const int output_cols = input_cols - patch_cols + 1;
150 output_cols);
178 output_k >= 0 && output_k < output_cols) {
204 const int output_cols = input_cols - patch_cols + 1;
212 Tensor<float, 4, RowMajor> output_backward(output_cols, output_rows,
241 output_k >= 0 && output_k < output_cols) {
266 const int output_cols = input_cols;
270 Tensor<float, 3> output_backward(output_depth, output_rows, output_cols);
292 output_j < output_cols) {
314 const int output_cols = input_cols;
320 Tensor<float, 3, RowMajor> output_backward(output_cols, output_rows,
343 output_j < output_cols) {
366 const int output_cols = input_cols;
375 output_cols);
411 output_k >= 0 && output_k < output_cols) {
437 const int output_cols = input_cols;
445 Tensor<float, 4, RowMajor> output_backward(output_cols, output_rows,
482 output_k >= 0 && output_k < output_cols) {
508 const int output_cols = input_cols - patch_cols + 1;
513 Tensor<float, 4> output_backward(output_depth, output_rows, output_cols,
539 output_j < output_cols) {
563 const int output_cols = input_cols - patch_cols + 1;
569 Tensor<float, 4, RowMajor> output_backward(num_batches, output_cols,
595 output_j < output_cols) {
620 const int output_cols = input_cols - patch_cols + 1;
629 output_cols, num_batches);
659 output_k >= 0 && output_k < output_cols) {
687 const int output_cols = input_cols - patch_cols + 1;
696 num_batches, output_cols, output_rows, output_planes, output_depth);
726 output_k >= 0 && output_k < output_cols) {
752 const int output_cols = divup(input_cols - patch_cols + 1, stride);
757 Tensor<float, 4> output_backward(output_depth, output_rows, output_cols,
783 output_j >= 0 && output_j / stride < output_cols &&
828 const int output_cols = divup(input_cols - patch_cols + 1, stride);
834 Tensor<float, 4, RowMajor> output_backward(num_batches, output_cols,
860 output_j >= 0 && output_j / stride < output_cols &&
904 const int output_cols = input_cols - patch_cols + 1;
908 Tensor<float, 3> output_backward(output_depth, output_rows, output_cols);
932 output_j < output_cols) {
954 const int output_cols = input_cols - patch_cols + 1;
959 Tensor<float, 3, RowMajor> output_backward(output_cols, output_rows,
984 output_j < output_cols) {
1013 const int output_cols = input_cols - patch_cols_eff + 1;
1015 Tensor<float, 4> output_backward(output_depth, output_rows, output_cols,
1063 const int output_cols = input_cols - patch_cols_eff + 1;
1065 Tensor<float, 4> output_backward(output_depth, output_rows, output_cols,
1110 const int output_cols = input_cols - patch_cols_eff + 1;
1112 Tensor<float, 4, RowMajor> output_backward(num_batches, output_cols,
1158 const int output_cols = input_cols - patch_cols_eff + 1;
1160 Tensor<float, 4> output_backward(output_depth, output_rows, output_cols,
1205 const int output_cols = input_cols - patch_cols_eff + 1;
1207 Tensor<float, 4, RowMajor> output_backward(num_batches, output_cols,
1248 const int output_cols = input_cols - patch_cols + 1;
1255 output_cols);
1284 output_k >= 0 && output_k < output_cols) {
1311 const int output_cols = input_cols - patch_cols + 1;
1318 Tensor<float, 4, RowMajor> output_backward(output_cols, output_rows,
1348 output_k >= 0 && output_k < output_cols) {
1374 const int output_cols = input_cols - patch_cols + 1;
1379 Tensor<float, 4> output_backward(output_depth, output_rows, output_cols,
1405 output_j < output_cols) {
1429 const int output_cols = input_cols - patch_cols + 1;
1435 Tensor<float, 4, RowMajor> output_backward(num_batches, output_cols,
1461 output_j < output_cols) {
1488 const int output_cols =
1495 Tensor<float, 4, RowMajor> output_backward(num_batches, output_cols,
1522 output_j / c_stride < output_cols &&
1550 const int output_cols = input_cols - patch_cols + 1;
1558 output_cols, num_batches);
1588 output_k >= 0 && output_k < output_cols) {
1617 const int output_cols = input_cols - patch_cols + 1;
1625 num_batches, output_cols, output_rows, output_planes, output_depth);
1655 output_k >= 0 && output_k < output_cols) {
1685 const int output_cols = (input_cols - patch_cols + 1 + stride - 1) / stride;
1690 Tensor<float, 4> output_backward(output_depth, output_rows, output_cols,
1716 output_j >= 0 && output_j / stride < output_cols &&
1745 const int output_cols = (input_cols - patch_cols + 1 + stride - 1) / stride;
1751 Tensor<float, 4, RowMajor> output_backward(num_batches, output_cols,
1777 output_j >= 0 && output_j / stride < output_cols &&
1810 const int output_cols = ceil_div(input_cols - patch_cols + 1, stride_cols);
1819 output_cols, num_batches);
1851 output_k >= 0 && output_k / stride_cols < output_cols &&
1889 const int output_cols = ceil_div(input_cols - patch_cols + 1, stride_cols);
1898 num_batches, output_cols, output_rows, output_planes, output_depth);
1930 output_k >= 0 && output_k / stride_cols < output_cols &&
1965 const int output_cols = ceil_div(input_cols - patch_cols + 1, stride_cols);
1975 output_cols, num_batches);
2007 output_k >= 0 && output_k / stride_cols < output_cols &&
2042 const int output_cols = ceil_div(input_cols - patch_cols + 1, stride_cols);
2052 num_batches, output_cols, output_rows, output_planes, output_depth);
2084 output_k >= 0 && output_k / stride_cols < output_cols &&