def func(*args_):
return functions.layer_normalization(*args_)
def func(*args_):
y = functions.layer_normalization(*args_)
return y * y
def func(*args_):
y = functions.layer_normalization(*args_, eps=self.eps)
return y * y
def func(x):
args_ = x, args[1], args[2]
return functions.layer_normalization(*args_)
def func(*args_):
return functions.layer_normalization(*args_, eps=self.eps)
def func(x):
args_ = x, args[1], args[2]
return functions.layer_normalization(*args_, eps=self.eps)
def func(*args_):
y = functions.layer_normalization(*args_)
return y * y
def func(*args_):
return functions.layer_normalization(*args_)
def forward(self, inputs, device):
x, gamma, beta = inputs
y = functions.layer_normalization(x, gamma, beta, eps=self.eps)
return y,
