Searched refs:out_depth_stride (Results 1 – 1 of 1) sorted by relevance
226 const int64 out_depth_stride = shard_rows * shard_cols * shard_stride; in operator ()() local229 const int64 out_depth_buf_base = od * out_depth_stride; in operator ()()230 const int64 out_depth_base = (od_start + od) * out_depth_stride; in operator ()()306 const int64 out_depth_stride = shard_rows * shard_cols * shard_stride; in operator ()() local307 const int64 coord_stride = out_depth_stride * args.out_depth; in operator ()()319 const int64 out_depth_base = od * out_depth_stride; in operator ()()773 const int64 out_depth_stride = filter_shards_row * filter_shards_col; in operator ()() local774 const int64 num_tiles_stride = args.out_depth * out_depth_stride; in operator ()()782 const int64 out_depth_base = od * out_depth_stride; in operator ()()