Searched refs:_validate_tensors_to_flatten (Results 1 – 1 of 1) sorted by relevance
634 ) = self._validate_tensors_to_flatten(params)757 def _validate_tensors_to_flatten( member in FlatParamHandle819 dtype, _, device = self._validate_tensors_to_flatten(tensors)