Searched refs:_test_embedding_model (Results 1 – 1 of 1) sorted by relevance
408 def _test_embedding_model(self, model, num_embedding, device): member in TestExpandedWeightFunctional597 return self._test_embedding_model(embedding_model, 16, device)