Searched refs:ShapedSlice (Results 1 – 6 of 6) sorted by relevance
37 InfeedThunk(ThunkInfo thunk_info, std::vector<ShapedSlice> dest_slices);45 const std::vector<ShapedSlice> dest_slices_;
35 OutfeedThunk(ThunkInfo thunk_info, std::vector<ShapedSlice> source_slices);43 const std::vector<ShapedSlice> source_slices_;
30 std::vector<ShapedSlice> dest_slices) in InfeedThunk()
29 std::vector<ShapedSlice> source_slices) in OutfeedThunk()
154 struct ShapedSlice { struct
3195 std::vector<ShapedSlice> dest_slices; in HandleInfeed()3201 dest_slices.push_back(ShapedSlice{slice, shape}); in HandleInfeed()3214 std::vector<ShapedSlice> source_slices; in HandleOutfeed()3220 source_slices.push_back(ShapedSlice{slice, shape}); in HandleOutfeed()