Searched refs:new_tensor_args (Results 1 – 2 of 2) sorted by relevance
1994 new_tensor_args = []2012 new_tensor_args.append(f"{arg.codegen_reference()}")2031 new_tensor_args.extend([f"{a.codegen_reference()}" for a in arg])2038 new_tensor_args.extend(2084 new_tensor_args.append(f"{arg}")2115 return new_tensor_args, new_int_args
4128 def unflatten_args(new_tensor_args, new_non_tensor_args): argument4130 it_tensors = iter(new_tensor_args)