Searched refs:_test_mixed_precision_embedding_table (Results 1 – 1 of 1) sorted by relevance
587 def _test_mixed_precision_embedding_table(self, mp_config): member in TestFSDPMixedPrecisionSharded621 self._test_mixed_precision_embedding_table(627 self._test_mixed_precision_embedding_table(641 self._test_mixed_precision_embedding_table(mp_config=default_mp_config)650 self._test_mixed_precision_embedding_table(