Searched defs:gathered_tensors (Results 1 – 2 of 2) sorted by relevance
29 def check_gathered_tensors(self, self_rank, size, gathered_tensors, assertFunc): argument
167 const std::vector<at::Tensor> gathered_tensors = output_tensors[i]; in verify_tensors() local