Searched refs:gen_script_fn_and_args (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | jit_metaprogramming_utils.py | 384 def gen_script_fn_and_args(method_name, func_type, *args, **kwargs): function
397 fn, tensors = gen_script_fn_and_args(method_name, func_type, *args, **kwargs)
525 script_fn, inputs = gen_script_fn_and_args(name, "nn_functional", *f_args_variable)
|