Searched refs:mock_wrapper (Results 1 – 2 of 2) sorted by relevance
17 TestCase, run_tests, freeze_rng_state, mock_wrapper, get_tensors_from, gradcheck,98 parameter_new = mock_wrapper(torch.nn.Parameter.__new__)100 register_buffer = mock_wrapper(torch.nn.Module.register_buffer)125 uninit_param_new = mock_wrapper(torch.nn.UninitializedParameter.__new__)127 uninit_buffer_new = mock_wrapper(torch.nn.UninitializedBuffer.__new__)
4871 def mock_wrapper(method): function