def test_relu_inplace(seed, ctx, func_name):
from nbla_test_utils import inplace_function_test_helper
x = nn.Variable([2, 3, 4], need_grad=True)
inplace_function_test_helper([x],
F.relu,
ctx=ctx,
rng=np.random.RandomState(seed))
def test_add2_inplace(seed, ctx, func_name):
from nbla_test_utils import inplace_function_test_helper
x0 = nn.Variable([2, 3, 4], need_grad=True)
x1 = nn.Variable([2, 3, 4], need_grad=True)
inplace_function_test_helper([x0, x1],
F.add2,
ctx=ctx,
rng=np.random.RandomState(seed))
def test_reshpae_inplace(seed, ctx, func_name, inshape, outshape):
from nbla_test_utils import inplace_function_test_helper
rng = np.random.RandomState(seed)
inputs = [nn.Variable.from_numpy_array(
rng.randn(*inshape).astype(np.float32))]
inplace_function_test_helper(
inputs, F.reshape, func_args=[
outshape], ctx=ctx, rng=np.random.RandomState(seed))
def test_add_scalar_inplace(seed, val, ctx, func_name):
from nbla_test_utils import inplace_function_test_helper
rng = np.random.RandomState(seed)
x = nn.Variable([2, 3, 4], need_grad=True)
inplace_function_test_helper([x],
F.add_scalar,
func_args=[val],
ctx=ctx,
rng=rng)
def test_transform_binary_inplace(seed, fname, ctx, func_name):
from nbla_test_utils import inplace_function_test_helper
x0 = nn.Variable([2, 3, 4], need_grad=True)
x1 = nn.Variable([2, 3, 4], need_grad=True)
func = getattr(F, fname)
inplace_function_test_helper([x0, x1],
func,
ctx=ctx,
rng=np.random.RandomState(seed))
def test_relu_inplace(seed, ctx, func_name):
from nbla_test_utils import inplace_function_test_helper
x = nn.Variable([2, 3, 4], need_grad=True)
inplace_function_test_helper(
[x], F.relu, ctx=ctx, rng=np.random.RandomState(seed))
