Searched refs:clone_if_tensor (Results 1 – 2 of 2) sorted by relevance
279 def clone_if_tensor(x): function299 batched_args = pytree.tree_map(clone_if_tensor, orig_batched_args)300 kwarg_values = pytree.tree_map(clone_if_tensor, orig_kwarg_values)
1070 return clone_if_tensor(arg)1072 ew_input = clone_if_tensor(sample_input.input)1147 def clone_if_tensor(t): function