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 …]
|