Searched refs:compute_result_and_gradient (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 8681 def compute_result_and_gradient(reduction, target_dtype): function
8699 result_long, grad_long = compute_result_and_gradient(reduction, torch.long)
8700 result_byte, grad_byte = compute_result_and_gradient(reduction, torch.uint8)
|
| H A D | test_nn.py | 11606 def compute_result_and_gradient(reduction, target_dtype): function
11618 result_long, grad_long = compute_result_and_gradient(reduction, torch.long)
11619 result_byte, grad_byte = compute_result_and_gradient(reduction, torch.uint8)
|