Searched defs:check_same_size (Results 1 – 3 of 3) sorted by relevance
6 def maybe_view(tensor, size, check_same_size=True): argument12 def maybe_unexpand(tensor, old_size, check_same_size=True): argument
374 bool check_same_size(const std::vector<at::Tensor>& input_tensors) { in check_same_size() function
2435 bool check_same_size(const std::vector<at::Tensor>& input_tensors) { in checkForNCCLErrorsInternal() function