Searched refs:use_record_function (Results 1 – 1 of 1) sorted by relevance
64 def udf_with_torch_ops(device=-1, use_record_function=False): argument68 if use_record_function2156 self, rpc_exec_mode, func, args, use_record_function=False, dst=None, kineto_profile=False argument2166 if not use_record_function2214 if use_record_function:2225 if use_record_function:2235 use_record_function=True)2251 use_record_function=True2265 use_record_function=True)2286 use_record_function=True[all …]