Searched refs:dim_metadata_ (Results 1 – 2 of 2) sorted by relevance
137 dim_metadata_.resize(num_expanded_dims * 2); in DenseToSparse()141 dim_metadata_[i * 2].clear(); in DenseToSparse()142 dim_metadata_[i * 2 + 1].clear(); in DenseToSparse()145 dim_metadata_[i * 2].push_back(expanded_shape[traversal_order_[i]]); in DenseToSparse()147 dim_metadata_[i * 2].push_back(0); // Segment array always begins with 0. in DenseToSparse()169 dim_metadata_[2 * dst_dim + 1].push_back(coordinate[dst_dim]); in DenseToSparse()187 int erase_offset = dim_metadata_[2 * dst_dim_idx + 1].size() * in DenseToSparse()190 auto& segments = dim_metadata_[2 * inner_compressed_dim[dst_dim_idx]]; in DenseToSparse()204 dim_metadata_[2 * dst_dim_idx].push_back( in DenseToSparse()205 dim_metadata_[2 * dst_dim_idx + 1].size()); in DenseToSparse()[all …]
82 return dim_metadata_; in GetDimMetadata()136 std::vector<std::vector<int>> dim_metadata_; variable