Searched refs:unique_data_ptrs (Results 1 – 2 of 2) sorted by relevance
207 std::unordered_set<void*> unique_data_ptrs; in flatten_parameters() local209 unique_data_ptrs.emplace(p.data_ptr()); in flatten_parameters()211 if (unique_data_ptrs.size() != flat_weights_.size()) { in flatten_parameters()
255 unique_data_ptrs = {p.data_ptr() for p in self._flat_weights}256 if len(unique_data_ptrs) != len(self._flat_weights):