Searched refs:shape_offset (Results 1 – 2 of 2) sorted by relevance
96 std::vector<int> shape_offset(num_original_dims); in DenseToSparse() local97 shape_offset[shape_offset.size() - 1] = 1; in DenseToSparse()99 shape_offset[i - 1] = shape_offset[i] * dense_shape_[i]; in DenseToSparse()104 expanded_shape_offset[i] = shape_offset[i]; in DenseToSparse()108 expanded_shape_offset[num_original_dims + i] = shape_offset[mapped_dim]; in DenseToSparse()
472 int64 shape_offset = 0; in Concat() local483 *ix_out++ = *st_ix++ + ((i % dims == primary_dim) ? shape_offset : 0); in Concat()488 shape_offset += st.shape()[primary_dim]; in Concat()