Searched refs:shard_cols (Results 1 – 1 of 1) sorted by relevance
200 const int64 shard_cols, const T* filter_in, in operator ()()226 const int64 out_depth_stride = shard_rows * shard_cols * shard_stride; in operator ()()234 for (int64 s_c = 0; s_c < shard_cols; ++s_c) { in operator ()()235 const int64 shard_base = shard_stride * (s_r * shard_cols + s_c); in operator ()()302 const int64 shard_cols = 1 + (residual_col + 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 ()()325 for (int64 s_c = 0; s_c < shard_cols; ++s_c) { in operator ()()330 const int64 shard_base = shard_stride * (s_r * shard_cols + s_c); in operator ()()[all …]