Searched defs:prefix_finite_grad_param (Results 1 – 2 of 2) sorted by relevance
194 torch::DeviceType device_type) { in TEST_F()230 torch::DeviceType device_type) { in TEST_F()279 auto prefix_finite_grad_param = std::get<1>(test_case); in TEST_F() local
12011 def gen_parameters(scalar, grad_only_one_elem, prefix_finite_grad_param): argument12028 …case(norm_type, error_if_nonfinite, scalar, grad_only_one_elem, prefix_finite_grad_param, is_norm_… argument