def test_general():
for condition_custom in [
{},
{"x_order": OrderNCHW}
]:
condition = dict(condition_default)
condition.update(condition_custom)
vx = np.random.rand(2, 3, 4, 5) - 0.5
vy = vx.copy()
vy[vx < 0] = np.exp(vy[vx < 0]) - 1
x = Variable(vx.shape, order=OrderNHWC)
y, = Elu(None)(x)
x.change_order(condition["x_order"])
y.change_order(condition["y_order"])
generate_kernel_test_case(
description=f"Elu: " + (", ".join([f"{k}={v}" for k, v in condition_custom.items()])),
backend=condition["backend"],
graph=Graph([x], [y]),
inputs={x: ConstantVariable(vx, OrderNHWC).change_order(x.order).data},
expected={y: ConstantVariable(vy, OrderNHWC).change_order(y.order).data},
raise_skip=False
)
raise SkipTest
def template(x_order=OrderNHWC, y_order=OrderNHWC, description: str = ""):
vx = np.random.rand(2, 3, 4, 5) - 0.5
vy = vx.copy()
vy[vx < 0] = np.exp(vy[vx < 0]) - 1
x = Variable(vx.shape, order=OrderNHWC)
y, = Elu(None)(x)
x.change_order(x_order)
y.change_order(y_order)
generate_kernel_test_case(
description=f"Elu {description}",
graph=Graph([x], [y]),
inputs={x: np.transpose(vx, [OrderNHWC.axes_dict[a] for a in x.order.axes])},
expected={y: np.transpose(vy, [OrderNHWC.axes_dict[a] for a in y.order.axes])},
)
