Searched refs:extract_tensor (Results 1 – 1 of 1) sorted by relevance
73 static inline at::Tensor extract_tensor(PyObject* obj);149 auto output = extract_tensor(_output); in THCPModule_nccl_reduce()292 static inline at::Tensor extract_tensor(PyObject* obj) { in extract_tensor() function