

def update_one_gpu(self, param, state):
    cuda.elementwise('T grad, T lr, T alpha, T eps', 'T param, T ms', 'ms = alpha * ms + (1 - alpha) * grad * grad;\n               param -= lr * grad / (sqrt(ms) + eps);', 'rmsprop')(param.grad, self.lr, self.alpha, self.eps, param.data, state['ms'])

dataset/ETHPy150Open pfnet/chainer/chainer/optimizers/rmsprop.py RMSprop.update_one_gpu/original