def create_dropout_states(handle):
state_size = cudnn.dropoutGetStatesSize(handle)
return cupy.empty((state_size,), dtype="b")
def __init__(self, handle, seed):
state_size = cudnn.dropoutGetStatesSize(handle)
self._states = memory.alloc(state_size)
self._desc = create_dropout_descriptor(handle, 0., self._states.ptr,
state_size, seed)
def create_dropout_states(handle):
state_size = cudnn.dropoutGetStatesSize(handle)
return cupy.empty((state_size,), dtype='b')
def create_dropout_states(handle):
warnings.warn(
'create_dropout_states is deprecated.'
'Please use DropoutStates class instead.', DeprecationWarning)
state_size = cudnn.dropoutGetStatesSize(handle)
return cupy.empty((state_size, ), dtype='b')
