Searched refs:shard_rows (Results 1 – 1 of 1) sorted by relevance
199 const int64 num_filters, const int64 shard_rows, in operator ()()226 const int64 out_depth_stride = shard_rows * shard_cols * shard_stride; in operator ()()233 for (int64 s_r = 0; s_r < shard_rows; ++s_r) { in operator ()()298 const int64 shard_rows = 1 + (residual_row + 2 - 1) / 2; in operator ()() local306 const int64 out_depth_stride = shard_rows * shard_cols * shard_stride; in operator ()()309 num_filters * shard_rows * shard_cols * args.in_depth; in operator ()()314 shard_rows * shard_cols * args.in_depth; in operator ()()322 for (int64 s_r = 0; s_r < shard_rows; ++s_r) { in operator ()()361 shard_rows, shard_cols, filter_buf, in operator ()()