Searched refs:_assert_interleaved_struct (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/autograd/ |
| H A D | test_functional.py | 119 def _assert_interleaved_struct(self, res, base1, base2): member in TestAutogradFunctional
706 self._assert_interleaved_struct(res, foo(inp), inp)
714 self._assert_interleaved_struct(res, foo(*inp), inp)
732 self._assert_interleaved_struct(res, foo(inp), inp)
741 self._assert_interleaved_struct(res, foo(inp), inp)
755 self._assert_interleaved_struct(res, inp, inp)
790 self._assert_interleaved_struct(res, exp_reducer(inputs), inputs)
798 self._assert_interleaved_struct(res, identity(inputs), inputs)
807 self._assert_interleaved_struct(res, add_exp_reducer(*inputs), inputs)
837 self._assert_interleaved_struct(res, exp_reducer(inputs), inputs)
[all …]
|