Home | History | Annotate | Download | only in kernels

Lines Matching defs:output_cols

35   const int output_cols = input_cols;
39 Tensor<float, 3> result(output_depth, output_rows, output_cols);
49 EXPECT_EQ(result.dimension(2), output_cols);
53 for (int j = 0; j < output_cols; ++j) {
59 c - 1 + j < output_cols) {
80 const int output_cols = input_cols;
85 Tensor<float, 3, RowMajor> result(output_cols, output_rows, output_depth);
92 EXPECT_EQ(result.dimension(0), output_cols);
98 for (int j = 0; j < output_cols; ++j) {
104 c - 1 + j < output_cols) {
200 const int output_cols = input_cols - patch_cols + 1;
204 Tensor<float, 4> result(output_depth, output_rows, output_cols, num_batches);
216 EXPECT_EQ(result.dimension(2), output_cols);
222 for (int j = 0; j < output_cols; ++j) {
255 const int output_cols = 1;
259 Tensor<float, 4> result(output_depth, output_rows, output_cols, num_batches);
271 EXPECT_EQ(result.dimension(2), output_cols);
278 for (int j = 0; j < output_cols; ++j) {
310 const int output_cols = input_cols - patch_cols + 1;
316 Tensor<float, 4, RowMajor> result(num_batches, output_cols, output_rows,
329 EXPECT_EQ(result.dimension(1), output_cols);
336 for (int j = 0; j < output_cols; ++j) {
366 const int output_cols = 2;
370 Tensor<float, 4> result(output_depth, output_rows, output_cols, num_batches);
382 EXPECT_EQ(result.dimension(2), output_cols);
388 for (int j = 0; j < output_cols; ++j) {
414 const int output_cols = 2;
418 Tensor<float, 4> result(output_depth, output_rows, output_cols, num_batches);
430 EXPECT_EQ(result.dimension(2), output_cols);
436 for (int j = 0; j < output_cols; ++j) {
462 const int output_cols = 2;
468 Tensor<float, 4, RowMajor> result(num_batches, output_cols, output_rows,
480 EXPECT_EQ(result.dimension(1), output_cols);
487 for (int j = 0; j < output_cols; ++j) {
513 const int output_cols = 3;
517 Tensor<float, 4> result(output_depth, output_rows, output_cols, num_batches);
531 EXPECT_EQ(result.dimension(2), output_cols);
537 for (int j = 0; j < output_cols; ++j) {
564 const int output_cols = 3;
570 Tensor<float, 4, RowMajor> result(num_batches, output_cols, output_rows,
584 EXPECT_EQ(result.dimension(1), output_cols);
591 for (int j = 0; j < output_cols; ++j) {
618 const int output_cols = 3;
624 Tensor<float, 4, RowMajor> result(num_batches, output_cols, output_rows,
640 EXPECT_EQ(result.dimension(1), output_cols);
647 for (int j = 0; j < output_cols; ++j) {