Searched defs:output_lists (Results 1 – 2 of 2) sorted by relevance
2073 std::vector<std::vector<at::Tensor>>& output_lists, in AsyncAllgatherCoalescedWork()2089 std::vector<std::vector<at::Tensor>> output_lists{}; member in c10d::__anon2d5ad4bf1311::AsyncAllgatherCoalescedWork2139 std::vector<std::vector<at::Tensor>>& output_lists, in allgather_coalesced()2199 std::vector<std::vector<at::Tensor>> output_lists(getSize()); in allgather_into_tensor_coalesced() local
597 PJRT_Buffer*** output_lists; // in/out member