Cross Reference: /external/tensorflow/tensorflow/core/kernels/depthwise_conv_op

Lines Matching refs:in_cols
44          args.in_cols <= 32 && args.in_rows == args.out_rows &&
45          args.in_cols == args.out_cols && args.pad_rows >= 0 &&
49              (args.in_rows + 1) / 2 * args.in_cols;
57          args.in_cols <= 32 && args.in_rows == args.out_rows &&
58          args.in_cols == args.out_cols && args.pad_rows >= 0 &&
61          args.filter_rows * args.filter_cols <= args.in_cols * block_height;
76   const int in_width = args.in_cols;
180   const int in_width = args.in_cols;
190   assert(blockDim.y == args.in_cols);
312   const int in_width = args.in_cols;
460   const int in_width = args.in_cols;
470   assert(blockDim.x == args.in_cols);
599       block_dim = dim3(kBlockDepth, args.in_cols, block_height);
608       block_dim = dim3(args.in_cols, block_height, kBlockDepth);
620   const int tile_width = args.in_cols + args.filter_cols - 1;
659   const int block_pixels = (args.in_rows + 1) / 2 * args.in_cols;
766   const int in_width = args.in_cols;
836   const int in_width = args.in_cols;
927       args.batch * args.in_rows * args.in_cols * args.in_depth;
992   const int in_width = args.in_cols;
1113 // kAccumPixels * 64 >= args.in_rows * args.in_cols * kBlockDepth.
1126   const int in_width = blockDim.y;  // slower (see b/62280718): args.in_cols;
1136   assert(blockDim.y == args.in_cols);
1272   const int in_width = args.in_cols;
1383 // kAccumPixels * 64 >= args.in_rows * args.in_cols * kBlockDepth.
1396   const int in_width = blockDim.x;  // slower (see b/62280718): args.in_cols;
1405   assert(blockDim.x == args.in_cols);
1534   const int tile_width = args.in_cols + args.filter_cols - 1;
1549       block_dim = dim3(kBlockDepth, args.in_cols, block_height);
1556       block_dim = dim3(args.in_cols, block_height, kBlockDepth);
1583   const int block_pixels = block_height * args.in_cols * kBlockDepth;
1612     // args.in_cols * block_height * kBlockDepth must be multiple of 32.
1613     for (; block_height * args.in_cols * block_depth & 31;
1617     int block_size = block_height * args.in_cols * block_depth;