Searched refs:dest_slices (Results 1 – 3 of 3) sorted by relevance
30 std::vector<ShapedSlice> dest_slices) in InfeedThunk() argument31 : Thunk(Kind::kInfeed, thunk_info), dest_slices_(std::move(dest_slices)) {} in InfeedThunk()
37 InfeedThunk(ThunkInfo thunk_info, std::vector<ShapedSlice> dest_slices);
3195 std::vector<ShapedSlice> dest_slices; in HandleInfeed() local3196 dest_slices.reserve(infeed_op.outputs().size()); in HandleInfeed()3201 dest_slices.push_back(ShapedSlice{slice, shape}); in HandleInfeed()3205 absl::make_unique<InfeedThunk>(input.thunk_info, std::move(dest_slices))); in HandleInfeed()