Searched refs:_vec_from_tensor_cpu (Results 1 – 1 of 1) sorted by relevance
1798 def _vec_from_tensor_cpu(*args): function1807 ur = _vec_from_tensor_cpu(inp, g_cpu, True)1810 ui = _vec_from_tensor_cpu(inp, g_cpu, True)1820 else [_vec_from_tensor_cpu(out, g_cpu) for out in outputs]