Searched refs:shard_stride (Results 1 – 1 of 1) sorted by relevance
225 const int64 shard_stride = args.in_depth; in operator ()() local226 const int64 out_depth_stride = shard_rows * shard_cols * shard_stride; in operator ()()235 const int64 shard_base = shard_stride * (s_r * shard_cols + s_c); in operator ()()305 const int64 shard_stride = args.in_depth; in operator ()() local306 const int64 out_depth_stride = shard_rows * shard_cols * shard_stride; in operator ()()330 const int64 shard_base = shard_stride * (s_r * shard_cols + s_c); in operator ()()