Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_load_state_dict.py469 if isinstance(src, MyWrapperLoadTensor):
484 if isinstance(src, MyWrapperLoadTensor):
535 class MyWrapperLoadTensor(MyLoadTensor): class
559 return t._data if isinstance(t, MyWrapperLoadTensor) else t
562 return MyWrapperLoadTensor(t) if isinstance(t, torch.Tensor) else t
604 subclasses = [None, MyLoadTensor, MyLoadTensor2, MyWrapperLoadTensor]