Home
last modified time | relevance | path

Searched refs:_assert_same_struct (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/autograd/
H A Dtest_functional.py100 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 …]