Searched refs:testCustomGradient (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | mirrored_variable_test.py | 613 def testCustomGradient(self, distribution): member in MirroredVariableCreationTest |
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ | ||
| H A D | forwardprop_test.py | 471 def testCustomGradient(self): member in ForwardpropTest |
| H A D | backprop_test.py | 1350 def testCustomGradient(self): member in BackpropTest |
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/ | ||
| H A D | while_v2_test.py | 105 def testCustomGradient(self): member in WhileV2Test |
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ | ||
| H A D | gradients_test.py | 1104 def testCustomGradient(self): member in CustomGradientTest |
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ | ||
| H A D | function_test.py | 252 def testCustomGradient(self): member in FunctionTest |