MCPcopy Create free account
hub / github.com/apache/singa / test_embedding_gpu

Method test_embedding_gpu

test/python/test_operation.py:3553–3554  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

3551
3552 @unittest.skipIf(not singa_wrap.USE_CUDA, 'CUDA is not enabled')
3553 def test_embedding_gpu(self):
3554 self.embedding_helper(gpu_dev)
3555
3556 @unittest.skipIf(not singa_wrap.USE_CUDA, 'CUDA is not enabled')
3557 def _cossim_value(self, dev=gpu_dev):

Callers

nothing calls this directly

Calls 1

embedding_helperMethod · 0.95

Tested by

no test coverage detected