Searched defs:test_backward (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_embedding.py | 1218 test_backward=True, argument
1344 test_backward=True, argument
|
| /aosp_15_r20/external/pytorch/test/quantization/core/experimental/ |
| H A D | test_fake_quantize.py | 82 def test_backward(self): member in TestFakeQuantize
|
| /aosp_15_r20/external/pytorch/benchmarks/fastrnns/ |
| H A D | test_bench.py | 45 def test_backward(self, modeldef, benchmark): member in TestBenchNetwork
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/normalization/ |
| H A D | layer_normalization_test.py | 317 def test_backward(self): member in LayerNormalizationNumericsTest
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_cpp_extensions_aot.py | 70 def test_backward(self): member in TestCppExtensionAOT
|
| H A D | test_flop_counter.py | 108 def test_backward(self): member in TestFlopCounter
|
| H A D | test_maskedtensor.py | 829 def test_backward(self): member in TestReductions
|
| H A D | test_ops.py | 1607 def test_backward(self, device, dtype, op): member in TestCompositeCompliance
|
| H A D | test_nestedtensor.py | 7316 def test_backward(self, device, dtype, op): member in TestNestedTensorOpInfo
|
| H A D | test_autograd.py | 2261 def test_backward(self): member in TestAutograd
|