Searched refs:shard_base (Results 1 – 1 of 1) sorted by relevance
235 const int64 shard_base = shard_stride * (s_r * shard_cols + s_c); in operator ()() local239 i * in_stride + out_depth_buf_base + shard_base; in operator ()()240 const int64 out_base = i * out_stride + out_depth_base + shard_base; in operator ()()330 const int64 shard_base = shard_stride * (s_r * shard_cols + s_c); in operator ()() local347 out_depth_base + shard_base; in operator ()()787 const int64 shard_base = sr * filter_shards_col + sc; in operator ()() local788 const int64 out_buf_base = tile_base + out_depth_base + shard_base; in operator ()()