Searched defs:set_qconfig (Results 1 – 2 of 2) sorted by relevance
1232 def test_embedding_api(self, num_embeddings, embedding_dim, set_qconfig): argument1271 def test_embedding_bag_api(self, num_embeddings, embedding_dim, num_offsets, set_qconfig): argument
1117 set_qconfig, is_emb_bag, dtype=torch.quint8): argument