Searched refs:traversal_order_ (Results 1 – 2 of 2) sorted by relevance
58 traversal_order_(traversal_order), in FormatConverter()114 dst_ordered_offset[i] = expanded_shape_offset[traversal_order_[i]]; in DenseToSparse()131 segment_count *= expanded_shape[traversal_order_[i]]; in DenseToSparse()143 dim_metadata_[i * 2].push_back(expanded_shape[traversal_order_[i]]); in DenseToSparse()176 int original_dim_idx = traversal_order_[dst_dim_idx]; in DenseToSparse()224 traversal_order_ = TfLiteIntArrayToVector(sparsity.traversal_order); in FormatConverter()248 int orig_dim = traversal_order_[original_rank + block_dim]; in FormatConverter()268 int orig_dim = traversal_order_[i]; in Populate()273 const int block_idx = traversal_order_[i] - orig_rank; in Populate()310 int total_rank = traversal_order_.size(); in SparseToDense()[all …]
90 std::vector<int> traversal_order_; variable