Searched refs:tensor_to_tensor_value (Results 1 – 1 of 1) sorted by relevance
1294 FastMap<at::Tensor*, const Value*> tensor_to_tensor_value; in checkStorageGroups() local1298 tensor_to_tensor_value.find(key_value.second), in checkStorageGroups()1299 tensor_to_tensor_value.end()); in checkStorageGroups()1300 tensor_to_tensor_value.emplace(key_value.second, key_value.first); in checkStorageGroups()1322 const auto* v1 = tensor_to_tensor_value.at(group[i]); in checkStorageGroups()1323 const auto* v2 = tensor_to_tensor_value.at(group[j]); in checkStorageGroups()