Searched refs:tensor_to_int_script (Results 1 – 1 of 1) sorted by relevance
257 def tensor_to_int_script(x, tensor): function262 tensor_to_int_script(torch.tensor([2]), torch.tensor([2, 2]))283 script_funs = [tensor_to_int_script, tensor_to_float_script]