Searched refs:tensor_set (Results 1 – 2 of 2) sorted by relevance
127 ska::flat_hash_set<AllocationID> tensor_set; in calculateUniqueTensorIDs() local131 tensor_set.insert(*t.allocation_id_ref_.get()); in calculateUniqueTensorIDs()138 [&tensor_set](const auto& i) { in calculateUniqueTensorIDs()139 auto it = tensor_set.find(*i.allocation_id_ref_.get()); in calculateUniqueTensorIDs()140 return it == tensor_set.end(); in calculateUniqueTensorIDs()
267 tensor_set = {275 self.assertLen(tensor_set, 4)276 self.assertIn(x1.ref(), tensor_set)277 self.assertIn(x2.ref(), tensor_set)278 self.assertIn(y.ref(), tensor_set)279 self.assertIn(z.ref(), tensor_set)280 self.assertIn(w.ref(), tensor_set)