Searched defs:test_grad (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/examples/adding_an_op/ | ||
| H A D | zero_out_2_test.py | 35 def test_grad(self): member in ZeroOut2Test |
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_modules.py | 515 def test_grad(self, device, dtype, module_info, training): member in TestModule |
| H A D | test_overrides.py | 588 def test_grad(self): member in TestTorchFunctionOverride |
| H A D | test_autograd.py | 900 def test_grad(self): member in TestAutograd |
| /aosp_15_r20/external/pytorch/test/functorch/ | ||
| H A D | test_ops.py | 492 def test_grad(self, device, dtype, op): member in TestOperators |
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/ | ||
| H A D | control_flow_ops_test.py | 1095 def test_grad(self): member in TensorArrayTest |
| /aosp_15_r20/external/pytorch/test/dynamo/ | ||
| H A D | test_higher_order_ops.py | 3901 def test_grad(self): member in FuncTorchHigherOrderOpTests |
| H A D | test_repros.py | 5344 def test_grad(self): member in ReproTests |
| H A D | test_misc.py | 3938 def test_grad(self): member in MiscTests |