Lines Matching refs:tile_cols
177 // [tile_rows, tile_cols, out_depth, shard_rows, shard_cols, in_depth]
210 const int64 tile_cols = transform->input_shape().cols;
211 const int64 tile_spatial_size = tile_rows * tile_cols;
270 // [tile_rows, tile_cols, out_depth, shard_rows, shard_cols, in_depth]
373 // [tile_rows, tile_cols, out_depth, shard_rows, shard_cols, in_depth]
385 const int64 tile_cols = transform->input_shape().cols;
386 const int64 tile_spatial_size = tile_rows * tile_cols;
454 // [tile_rows, tile_cols, out_depth, shard_rows, shard_cols, in_depth]
625 // [tile_rows, tile_cols, num_tiles, in_depth]
637 const int64 tile_cols = transform->input_shape().cols;
649 for (int64 c = 0; c < tile_cols; ++c) {
677 // [tile_rows, tile_cols, num_tiles, in_depth]
681 // [tile_rows, tile_cols, num_tiles, in_depth]
699 const int64 tile_cols = transform->input_shape().cols;
700 const int64 tile_spatial_size = tile_rows * tile_cols;
726 // [tile_rows, tile_cols, num_tiles, out_depth, shard_rows, shard_cols]
751 const int64 tile_cols = transform->input_shape().cols;
752 const int64 tile_spatial_size = tile_rows * tile_cols;
946 const int64 tile_cols = transform->input_shape().cols;
947 const int64 tile_spatial_size = tile_rows * tile_cols;
968 TensorShape({tile_rows, tile_cols, out_depth,
1005 out_depth, tile_rows, tile_cols, out_tile_rows, out_tile_cols,
