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')