Searched refs:_nested_map (Results 1 – 4 of 4) sorted by relevance
295 def _nested_map(condition, fn, condition_msg=None): function361 randomize_args = _nested_map(is_tensor, randomize_arg)
654 def _nested_map(condition, fn, condition_msg=None): function758 _map_tensor_data = _nested_map(
5 from torch.autograd.function import _nested_map59 return _nested_map(lambda t: isinstance(t, torch.Tensor), fn)(input)
172 return function._nested_map(