Searched refs:assert_module_parameters_are (Results 1 – 1 of 1) sorted by relevance
3701 def assert_module_parameters_are(tensor_type, device_id=None): function3713 assert_module_parameters_are(torch.cuda.FloatTensor, 0) # type: ignore[attr-defined]3720 … assert_module_parameters_are(torch.cuda.FloatTensor, 1) # type: ignore[attr-defined]3745 assert_module_parameters_are(torch.FloatTensor)3751 assert_module_parameters_are(torch.DoubleTensor)3761 … assert_module_parameters_are(torch.cuda.FloatTensor, 0) # type: ignore[attr-defined]3767 assert_module_parameters_are(torch.FloatTensor)3773 … assert_module_parameters_are(torch.cuda.FloatTensor, 0) # type: ignore[attr-defined]3779 … assert_module_parameters_are(torch.cuda.FloatTensor, 0) # type: ignore[attr-defined]3788 … assert_module_parameters_are(torch.cuda.FloatTensor, 1) # type: ignore[attr-defined][all …]