def test_grad_image(func, motion, optimized, preserve_result, timage, tkernel,
conv2dstrides):
"""Test gradients of image functions."""
# TODO: Upgrade utils.py to allow simultaneous testing of uneven args.
tfe_utils.test_rev_tensor(func, motion, optimized, preserve_result, (0,),
timage, tkernel, conv2dstrides)
tfe_utils.test_rev_tensor(func, motion, optimized, preserve_result, (1,),
timage, tkernel, conv2dstrides)
def test_grad_image_pooling(func, motion, optimized, preserve_result, timage,
pool2dsizes, conv2dstrides):
tfe_utils.test_rev_tensor(func, motion, optimized, preserve_result, (0,),
timage, pool2dsizes, conv2dstrides)
def test_grad_tensor_broadcast(func, motion, optimized, preserve_result, s,
t):
"""Test gradients of functions with binary tensor inputs."""
tfe_utils.test_rev_tensor(func, motion, optimized, preserve_result, (0, 1),
s, t)
def test_grad_matmul_higherdim(func, motion, optimized, preserve_result,
timage1, timage2, boolean1, boolean2):
"""Test gradients of functions with binary matrix inputs."""
tfe_utils.test_rev_tensor(func, motion, optimized, preserve_result, (0, 1),
timage1, timage2, boolean1, boolean2)
def test_grad_binary_tensor(func, motion, optimized, preserve_result, t1, t2):
"""Test gradients of functions with binary tensor inputs."""
tfe_utils.test_rev_tensor(func, motion, optimized, preserve_result, (0, 1),
t1, t2)
def test_grad_unary_reduction(func, motion, optimized, preserve_result,
timage, boolean):
"""Test gradients of reduction functions."""
tfe_utils.test_rev_tensor(func, motion, optimized, preserve_result, (0,),
timage, boolean)
def test_grad_unary_tensor(func, motion, optimized, preserve_result, t): """Test gradients of functions with single tensor input.""" tfe_utils.test_rev_tensor(func, motion, optimized, preserve_result, (0,), t)