Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_nn.py3701 def assert_module_parameters_are(tensor_type, device_id=None): function
3713 assert_module_parameters_are(torch.cuda.FloatTensor, 0) # type: ignore[attr-defined]
3720assert_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)
3761assert_module_parameters_are(torch.cuda.FloatTensor, 0) # type: ignore[attr-defined]
3767 assert_module_parameters_are(torch.FloatTensor)
3773assert_module_parameters_are(torch.cuda.FloatTensor, 0) # type: ignore[attr-defined]
3779assert_module_parameters_are(torch.cuda.FloatTensor, 0) # type: ignore[attr-defined]
3788assert_module_parameters_are(torch.cuda.FloatTensor, 1) # type: ignore[attr-defined]
[all …]