Searched defs:wait_tensor (Results 1 – 2 of 2) sorted by relevance
299 at::Tensor wait_tensor(const at::Tensor& tensor) { in wait_tensor() function
134 def wait_tensor(tensor): function