pytorch
0c9dce90 - [pytorch] add per_sample_weights support for embedding_bag_4bit_rowwise_offsets (#63605)

Commit
3 years ago
[pytorch] add per_sample_weights support for embedding_bag_4bit_rowwise_offsets (#63605) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/63605 Reviewed By: houseroad Differential Revision: D30434664 fbshipit-source-id: eb4cbae3c705f9dec5c073a56f0f23daee353bc1
Parents
Loading