Home
last modified time | relevance | path

Searched defs:use_tape (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/
H A Dgradient_correctness_test.py31 def testMultipleOutputChainedGradients(self, use_tape): argument
45 def testIdentityGradient(self, use_tape): argument
53 def testIntegerIdentityGradient(self, use_tape): argument
61 def testGradientWithIntegerPath(self, use_tape): argument
72 def testNoIntegerGradient1(self, use_tape): argument
83 def testNoIntegerGradient2(self, use_tape): argument
94 def testNoIntegerGradient3(self, use_tape): argument
104 def testNoIntegerGradient4(self, use_tape): argument
114 def testNoIntegerGradient5(self, use_tape): argument
125 def testNoIntegerGradient6(self, use_tape): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
H A Dscatter_nd_ops_test.py649 def testGradientsRank2ElementUpdate(self, use_tape): argument
672 def testGradientsRank2SliceUpdate(self, use_tape): argument
695 def testGradientsRank3SliceUpdate(self, use_tape): argument
721 def testGradientsRank7SliceUpdate(self, use_tape): argument
H A Darray_ops_test.py1032 def __init__(self, test, var, varnp, use_tape): argument
1079 def testGradient(self, use_tape): argument
1107 def testGradientZero(self, use_tape): argument
1115 def testInt64Indices(self, use_tape): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dimage_grad_test_base.py154 def _testShapesParameterized(self, use_tape): argument
222 def testGradOnUnsupportedType(self, use_tape): argument
335 def testGradOnUnsupportedType(self, use_tape): argument
H A Dgradients_test.py1609 def testFnRecomputeWithScopeGradient(self, use_tape): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/image_ops/
H A Dextract_volume_patches_grad_test.py82 def testConstructGradientWithLargeVolumes(self, use_tape): argument
H A Dextract_image_patches_grad_test.py113 def testConstructGradientWithLargeImages(self, use_tape): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dtest_util.py3865 def __init__(self, use_tape, persistent=False): argument