Searched defs:WorkEntry (Results 1 – 1 of 1) sorted by relevance
31 struct WorkEntry { struct32 explicit WorkEntry( in WorkEntry() function48 std::vector<at::Tensor> src;51 const std::vector<at::Tensor> dst;54 int* srcRank = nullptr;55 std::function<void(std::unique_ptr<WorkEntry>&)> run;