Searched refs:expanded_shape_offset (Results 1 – 1 of 1) sorted by relevance
104 std::vector<int> expanded_shape_offset(num_expanded_dims); in DenseToSparse() local106 expanded_shape_offset[i] = shape_offset[i]; in DenseToSparse()110 expanded_shape_offset[num_original_dims + i] = shape_offset[mapped_dim]; in DenseToSparse()111 expanded_shape_offset[mapped_dim] *= block_size_[i]; in DenseToSparse()116 dst_ordered_offset[i] = expanded_shape_offset[traversal_order_[i]]; in DenseToSparse()