Searched defs:error_if_nonfinite (Results 1 – 2 of 2) sorted by relevance
230 torch::DeviceType device_type) { in TEST_F()
12028 …def run_test_case(norm_type, error_if_nonfinite, scalar, grad_only_one_elem, prefix_finite_grad_pa… argument