Xing Wang 73ae23a4ea add support for real4bits quant (#25426) ...

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/25426

Add embedding table 4bit quantization support.

* add the conversion from fp32 to int4.
* using brew to pass the context so that the 4bit operators are added when generating the predictor net.

Reviewed By: kennyhorror, chocjy

Differential Revision: D16859892

fbshipit-source-id: a06c3f0b56a7eabf9ca4a2b2cb6c63735030d70b

2019-09-20 13:45:23 -07:00

..

__init__.py

…

adaptive_weight.py

…

add_bias.py

…

arc_cosine_feature_map.py

…

batch_huber_loss.py

Add new regression loss function type to FBLearner (#21080)

2019-06-17 17:43:00 -07:00

batch_lr_loss.py

Support focal loss in MTML

2019-08-25 01:42:25 -07:00

batch_mse_loss.py

…

batch_normalization.py

…

batch_sigmoid_cross_entropy_loss.py

…

batch_softmax_loss.py

…

blob_weighted_sum.py

…

bpr_loss.py

Add BPR loss to TTSN (#24439)

2019-08-15 23:20:15 -07:00

bucket_weighted.py

Make hashing default for bucket-weighted pooling (#24266)

2019-08-13 13:56:32 -07:00

build_index.py

…

concat.py

…

constant_weight.py

…

conv.py

…

dropout.py

…

fc_without_bias.py

…

fc.py

…

feature_sparse_to_dense.py

Return list of AccessedFeatures from get_accessed_features (#23983)

2019-08-14 10:50:27 -07:00

functional.py

…

gather_record.py

…

homotopy_weight.py

…

label_smooth.py

…

last_n_window_collector.py

…

layer_normalization.py

…

layers.py

Return list of AccessedFeatures from get_accessed_features (#23983)

2019-08-14 10:50:27 -07:00

margin_rank_loss.py

…

merge_id_lists.py

…

pairwise_similarity.py

…

position_weighted.py

…

random_fourier_features.py

…

reservoir_sampling.py

…

sampling_train.py

…

sampling_trainable_mixin.py

…

select_record_by_context.py

…

semi_random_features.py

…

sparse_dropout_with_replacement.py

hook up dropout sparse with replacement operator

2019-07-23 14:34:25 -07:00

sparse_feature_hash.py

Refactor and expose metadata of tum_history layer for online prediction

2019-08-15 00:27:11 -07:00

sparse_lookup.py

add support for real4bits quant (#25426)

2019-09-20 13:45:23 -07:00

split.py

Enable variable size embedding (#25782)

2019-09-09 22:08:32 -07:00

tags.py

…

uniform_sampling.py

…