| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | gradient_correctness_test.py | 31 def testMultipleOutputChainedGradients(self, use_tape): argument
32 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
45 def testIdentityGradient(self, use_tape): argument
47 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
53 def testIntegerIdentityGradient(self, use_tape): argument
55 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
61 def testGradientWithIntegerPath(self, use_tape): argument
62 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
72 def testNoIntegerGradient1(self, use_tape): argument
73 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/image_ops/ |
| D | extract_volume_patches_grad_test.py | 82 def testConstructGradientWithLargeVolumes(self, use_tape): argument
83 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
|
| D | extract_image_patches_grad_test.py | 113 def testConstructGradientWithLargeImages(self, use_tape): argument
114 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_grad_test_base.py | 154 def _testShapesParameterized(self, use_tape): argument
162 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
222 def testGradOnUnsupportedType(self, use_tape): argument
226 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
335 def testGradOnUnsupportedType(self, use_tape): argument
336 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
|
| D | gradients_test.py | 1609 def testFnRecomputeWithScopeGradient(self, use_tape): argument
1628 use_tape=use_tape, persistent=True) as tape:
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | scatter_nd_ops_test.py | 649 def testGradientsRank2ElementUpdate(self, use_tape): argument
651 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
672 def testGradientsRank2SliceUpdate(self, use_tape): argument
674 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
695 def testGradientsRank3SliceUpdate(self, use_tape): argument
697 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
721 def testGradientsRank7SliceUpdate(self, use_tape): argument
723 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
|
| D | array_ops_test.py | 1032 def __init__(self, test, var, varnp, use_tape): argument
1036 self.use_tape = use_tape
1040 use_tape=self.use_tape, persistent=True) as tape:
1079 def testGradient(self, use_tape): argument
1087 grad = GradSliceChecker(self, var, raw, use_tape)
1107 def testGradientZero(self, use_tape): argument
1111 grad = GradSliceChecker(self, var, np.array(8), use_tape)
1115 def testInt64Indices(self, use_tape): argument
1116 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 3865 def __init__(self, use_tape, persistent=False): argument
3866 self._use_tape = use_tape
|