Searched refs:extract_tensors (Results 1 – 1 of 1) sorted by relevance
74 static inline std::vector<at::Tensor> extract_tensors(PyObject* obj);148 std::vector<at::Tensor> inputs = extract_tensors(_inputs); in THCPModule_nccl_reduce()182 std::vector<at::Tensor> inputs = extract_tensors(_inputs); in THCPModule_nccl_all_reduce()183 std::vector<at::Tensor> outputs = extract_tensors(_outputs); in THCPModule_nccl_all_reduce()213 std::vector<at::Tensor> inputs = extract_tensors(_inputs); in THCPModule_nccl_broadcast()245 std::vector<at::Tensor> inputs = extract_tensors(_inputs); in THCPModule_nccl_all_gather()246 std::vector<at::Tensor> outputs = extract_tensors(_outputs); in THCPModule_nccl_all_gather()278 std::vector<at::Tensor> inputs = extract_tensors(_inputs); in THCPModule_nccl_reduce_scatter()279 std::vector<at::Tensor> outputs = extract_tensors(_outputs); in THCPModule_nccl_reduce_scatter()301 static inline std::vector<at::Tensor> extract_tensors(PyObject* obj) { in extract_tensors() function