Searched refs:traversal_order_ (Results 1 – 2 of 2) sorted by relevance
60 traversal_order_(traversal_order), in FormatConverter()116 dst_ordered_offset[i] = expanded_shape_offset[traversal_order_[i]]; in DenseToSparse()133 segment_count *= expanded_shape[traversal_order_[i]]; in DenseToSparse()145 dim_metadata_[i * 2].push_back(expanded_shape[traversal_order_[i]]); in DenseToSparse()178 int original_dim_idx = traversal_order_[dst_dim_idx]; in DenseToSparse()260 traversal_order_ = std::move(traversal_order); in InitSparseToDenseConverter()286 if (original_rank + block_dim < traversal_order_.size()) { in InitSparseToDenseConverter()287 int orig_dim = traversal_order_[original_rank + block_dim]; in InitSparseToDenseConverter()308 int orig_dim = traversal_order_[i]; in Populate()313 const int block_idx = traversal_order_[i] - orig_rank; in Populate()[all …]
124 std::vector<int> traversal_order_; variable