pytorch/caffe2/python/layers
Jiyan Yang 0199d59d3a Resubmit: Set the correct engine name for position weighted pooling when fp16 is used for training
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/13768

Reviewed By: xianjiec

Differential Revision: D12996103

fbshipit-source-id: 5ca4cda4210f68ece2b5d6eced8cf52ee91fb36f
2018-11-27 14:51:56 -08:00
..
__init__.py
adaptive_weight.py
add_bias.py
arc_cosine_feature_map.py
batch_distill_lr_loss.py
batch_lr_loss.py
batch_mse_loss.py
batch_normalization.py
batch_sigmoid_cross_entropy_loss.py
batch_softmax_loss.py
blob_weighted_sum.py
bucket_weighted.py Implement bucket-based attention pooling for IdScoreList features (#13004) 2018-10-25 18:04:08 -07:00
build_index.py
concat.py
constant_weight.py
conv.py
dropout.py
fc.py fc layer accept axis argument (#13822) 2018-11-11 13:44:57 -08:00
fc_without_bias.py
feature_sparse_to_dense.py
functional.py
gather_record.py
homotopy_weight.py
label_smooth.py
last_n_window_collector.py
layer_normalization.py
layers.py
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_feature_hash.py
sparse_lookup.py Resubmit: Set the correct engine name for position weighted pooling when fp16 is used for training 2018-11-27 14:51:56 -08:00
split.py
tags.py
uniform_sampling.py