Searched refs:_assert_same_struct (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/autograd/ |
| H A D | test_functional.py | 100 def _assert_same_struct(self, res, base): member in TestAutogradFunctional
200 self._assert_same_struct(res, inp)
219 self._assert_same_struct(res[1], inp)
228 self._assert_same_struct(res[1], inp)
242 self._assert_same_struct(res[1], inp)
274 self._assert_same_struct(res[1], inputs)
284 self._assert_same_struct(vjp_val, inputs)
295 self._assert_same_struct(vjp_val, inputs)
309 self._assert_same_struct(res[0], v)
310 self._assert_same_struct(res[1], inputs)
[all …]
|