Module test_torch

distributed.protocol.tests.test_torch

Functions

▷ def test_deserialize_grad()
source link
▷ def test_grad(requires_grad) @pytest.mark.parametrize("requires_grad", [True, False])
@pytest.mark.parametrize("requires_grad", [True, False])
def test_grad(requires_grad)
source link
▷ def test_resnet()
source link
▷ def test_tensor()
source link