Home | History | Annotate | Download | only in kernels

Lines Matching defs:output_cols

36   const int output_cols = 2;
39 Tensor<float, 4> result(depth, output_rows, output_cols, num_batches);
51 EXPECT_EQ(result.dimension(2), output_cols);
57 for (int j = 0; j < output_cols; ++j) {
84 const int output_cols = 2;
87 Tensor<float, 4, RowMajor> result(num_batches, output_cols, output_rows,
100 EXPECT_EQ(result.dimension(1), output_cols);
106 for (int j = 0; j < output_cols; ++j) {
135 const int output_cols = 3;
140 Tensor<float, 5> result(channels, output_planes, output_rows, output_cols,
154 EXPECT_EQ(result.dimension(3), output_cols);
161 for (int k = 0; k < output_cols; ++k) {
195 const int output_cols = 3;
200 Tensor<float, 5, RowMajor> result(num_batches, output_cols, output_rows,
214 EXPECT_EQ(result.dimension(1), output_cols);
221 for (int k = 0; k < output_cols; ++k) {
255 const int output_cols = 3;
260 Tensor<float, 5> result(channels, output_planes, output_rows, output_cols,
274 EXPECT_EQ(result.dimension(3), output_cols);
281 for (int k = 0; k < output_cols; ++k) {
317 const int output_cols = 3;
322 Tensor<float, 5, RowMajor> result(num_batches, output_cols, output_rows,
336 EXPECT_EQ(result.dimension(1), output_cols);
343 for (int k = 0; k < output_cols; ++k) {
379 const int output_cols = input_cols;
384 Tensor<float, 5> result(channels, output_planes, output_rows, output_cols,
398 EXPECT_EQ(result.dimension(3), output_cols);
403 const int pad_c = output_cols - input_cols + patch_cols - 1;
415 for (int k = 0; k < output_cols; ++k) {
457 const int output_cols = input_cols;
462 Tensor<float, 5, RowMajor> result(num_batches, output_cols, output_rows,
476 EXPECT_EQ(result.dimension(1), output_cols);
481 const int pad_c = output_cols - input_cols + patch_cols - 1;
493 for (int k = 0; k < output_cols; ++k) {
533 const int output_cols = 2;
536 Tensor<float, 4> result(depth, output_rows, output_cols, num_batches);
547 EXPECT_EQ(result.dimension(2), output_cols);
553 for (int j = 0; j < output_cols; ++j) {
581 const int output_cols = 2;
584 Tensor<float, 4, RowMajor> result(num_batches, output_cols, output_rows,
596 EXPECT_EQ(result.dimension(1), output_cols);
602 for (int j = 0; j < output_cols; ++j) {
633 const int output_cols = 2;
637 Tensor<float, 5> result(channels, output_planes, output_rows, output_cols,
650 EXPECT_EQ(result.dimension(3), output_cols);
657 for (int k = 0; k < output_cols; ++k) {
693 const int output_cols = 2;
697 Tensor<float, 5, RowMajor> result(num_batches, output_cols, output_rows,
710 EXPECT_EQ(result.dimension(1), output_cols);
717 for (int k = 0; k < output_cols; ++k) {