Searched refs:primary_dim (Results 1 – 1 of 1) sorted by relevance
506 const int primary_dim = order_0[0]; in Concat() local509 final_shape[primary_dim] = 0; // We'll build this up as we go along. in Concat()518 DCHECK_EQ(st.order()[0], primary_dim) in Concat()523 const int cdim = (d < primary_dim) ? d : d + 1; in Concat()526 << "Concat dim: " << primary_dim in Concat()533 final_shape[primary_dim] = in Concat()534 (final_shape[primary_dim] + st_shape[primary_dim]); in Concat()561 *ix_out++ = *st_ix++ + ((i % dims == primary_dim) ? shape_offset : 0); in Concat()565 shape_offset += st.shape()[primary_dim]; in Concat()