Home
last modified time | relevance | path

Searched refs:use_tape (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dgradient_correctness_test.py35 def testMultipleOutputChainedGradients(self, use_tape): argument
36 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
49 def testIdentityGradient(self, use_tape): argument
51 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
57 def testIntegerIdentityGradient(self, use_tape): argument
59 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
65 def testGradientWithIntegerPath(self, use_tape): argument
66 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
76 def testNoIntegerGradient1(self, use_tape): argument
77 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
[all …]
Dextract_volume_patches_grad_test.py86 def testConstructGradientWithLargeVolumes(self, use_tape): argument
87 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
Dextract_image_patches_grad_test.py117 def testConstructGradientWithLargeImages(self, use_tape): argument
118 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
/external/tensorflow/tensorflow/python/ops/
Dimage_grad_test_base.py156 def _testShapesParameterized(self, use_tape): argument
164 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
224 def testGradOnUnsupportedType(self, use_tape): argument
228 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
337 def testGradOnUnsupportedType(self, use_tape): argument
338 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
Dgradients_test.py1555 def testFnRecomputeWithScopeGradient(self, use_tape): argument
1574 use_tape=use_tape, persistent=True) as tape:
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dscatter_nd_ops_test.py633 def testGradientsRank2ElementUpdate(self, use_tape): argument
635 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
656 def testGradientsRank2SliceUpdate(self, use_tape): argument
658 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
679 def testGradientsRank3SliceUpdate(self, use_tape): argument
681 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
705 def testGradientsRank7SliceUpdate(self, use_tape): argument
707 with test_util.AbstractGradientTape(use_tape=use_tape) as tape:
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py3443 def __init__(self, use_tape, persistent=False): argument
3444 self._use_tape = use_tape