which version is better?

eswish.py

from scipy.special import expit

class Eswish1(Layer):    
    def __init__(self, beta):
        self.ReLUness = 0
        self.layertype = 'Eswish'
        self.beta = beta
    
    def forward(self, input, is_train):
        self.input = input
        self.sigmoid = expit(input)
        self.fx = self.beta * input * self.sigmoid
        return self.fx
    
    def backward(self, dy):
        grad = self.fx + self.sigmoid * (self.beta - self.fx)
        return dy*grad
        
        
class Eswish2(Layer):    
    def __init__(self, beta):
        self.ReLUness = 0
        self.layertype = 'Eswish'
        self.beta = beta
    
    def forward(self, input, is_train):
        self.input = input
        sigmoid = expit(input)
        fx = self.beta * input * sigmoid
        self.grad = fx + sigmoid * (self.beta - fx)
        return fx
    
    def backward(self, dy):
        return dy*self.grad