コード例 #1
0
 def set_stream(self, stream=None):
     if stream is None:
         stream = cuda.Stream()
     curand.setStream(self._generator, stream.ptr)
コード例 #2
0
ファイル: generator.py プロジェクト: 2php/chainer 
 def set_stream(self, stream=None):
     if stream is None:
         stream = cuda.Stream()
     curand.setStream(self._generator, stream.ptr)