Searched defs:grad_out1 (Results 1 – 2 of 2) sorted by relevance
131 def backward(ctx, grad_out1, grad_out2): argument
2582 auto grad_out1 = at::rand({22, 22, 22}); in TEST() local