예제 #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)