Searched refs:_validate_output_tensor_for_gather (Results 1 – 2 of 2) sorted by relevance
53 def _validate_output_tensor_for_gather( function
36 _validate_output_tensor_for_gather,416 _validate_output_tensor_for_gather(rank, dst, full_size, out)