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

Method test_softmax_api_gpu

test/python/test_api.py:247–248  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

245
246 @unittest.skipIf(not singa_api.USE_CUDA, 'CUDA is not enabled')
247 def test_softmax_api_gpu(self):
248 self._softmax_api_helper(gpu_dev)
249
250 def _tensor_arithmetic_op_broadcast_helper(self, dev):
251

Callers

nothing calls this directly

Calls 1

_softmax_api_helperMethod · 0.95

Tested by

no test coverage detected