Searched refs:tensor_or_tuple_of_tensors (Results 1 – 1 of 1) sorted by relevance
141 def _any_differentiable(tensor_or_tuple_of_tensors): argument142 flat_args, _ = tree_unflatten(tensor_or_tuple_of_tensors)