Lines Matching full:requires_grad_

45             cpu_tensor.requires_grad_()
204         root = torch.randn(4, 5, dtype=torch.float32).to_mkldnn().requires_grad_()
216         root = torch.randn(4, 5, dtype=torch.float32).to_mkldnn().requires_grad_()
256                 x1.requires_grad_()
257                 x2.requires_grad_()
332                 x_ref = x_lower.clone().float().detach().requires_grad_()
333                 x_lower.requires_grad_()
384                 x1.requires_grad_()
385                 x2.requires_grad_()
498                 x.requires_grad_()
499                 x_ref.requires_grad_()
567         x1 = x.clone().requires_grad_()
568         x2 = x.clone().to_mkldnn().requires_grad_()
580         x1 = x.clone().requires_grad_()
581         x2 = x.clone().to_mkldnn().requires_grad_()
615         x1 = x.clone().requires_grad_()
616         x2 = x.clone().to_mkldnn().requires_grad_()
630         x1 = x.clone().to_mkldnn().requires_grad_()
631         x2 = x.clone().to_mkldnn(torch.bfloat16).requires_grad_()
649         x1 = x.clone().requires_grad_()
650         x2 = x.clone().to_mkldnn().requires_grad_()
651         x3 = x.clone().to_mkldnn().requires_grad_()
684             x_fp32 = x.clone().to_mkldnn().requires_grad_()
685             x_bf16 = x.clone().to_mkldnn(torch.bfloat16).requires_grad_()
699             x_bf16 = torch.randn(size, dtype=torch.bfloat16).requires_grad_()
725                 x1 = input.clone().requires_grad_()
726                 x2 = input.clone().to_mkldnn().requires_grad_()
848             x1 = input.clone().requires_grad_()
849             x2 = input.clone().to_mkldnn().requires_grad_()
930         x1 = x.clone().requires_grad_()
931         x2 = x.clone().to_mkldnn().requires_grad_()
984             x1 = input.clone().requires_grad_()
985             x2 = input.clone().to_mkldnn().requires_grad_()
1149         x.requires_grad_(True)
1150         y.requires_grad_(True)
1208         x1 = x.clone().requires_grad_()
1209         x2 = x.clone().to_mkldnn().requires_grad_()
1254             x1 = x.clone().requires_grad_()
1255             x2 = x.clone().to_mkldnn().requires_grad_()
1288             x1 = x.clone().requires_grad_()
1289             x2 = x.clone().to_mkldnn().requires_grad_()
1511                 input1 = input.clone().requires_grad_(training)
1512                 input2 = input.clone().requires_grad_(training)
1514                 h1 = h.clone().requires_grad_(training)
1515                 h2 = h.clone().requires_grad_(training)
1516                 c1 = c.clone().requires_grad_(training)
1517                 c2 = c.clone().requires_grad_(training)