pytorch/caffe2/python/layers
Bangsheng Tang d8443b8ffa BatchGatherOp
Summary:
1. added BatchGatherOp and BatchGatherGradientOp
2. unit tests

Reviewed By: xianjiec

Differential Revision: D5443965

fbshipit-source-id: bdcbb7f9f91c55484372a4bdb1727ae6d49e2018
2017-07-27 10:17:42 -07:00
..
__init__.py Allow to import subclasses of layers 2017-07-12 20:19:47 -07:00
add_bias.py
arc_cosine_feature_map.py Modify arc cosine feature map and semi random layers to initialize parameters as global constants 2017-07-26 16:37:00 -07:00
batch_distill_lr_loss.py Feature importance in dper 2.0: build network representation 2017-06-05 18:03:34 -07:00
batch_lr_loss.py Feature importance in dper 2.0: build network representation 2017-06-05 18:03:34 -07:00
batch_mse_loss.py Feature importance in dper 2.0: build network representation 2017-06-05 18:03:34 -07:00
batch_normalization.py implement drelu and unittest 2017-07-20 11:50:08 -07:00
batch_sigmoid_cross_entropy_loss.py Feature importance in dper 2.0: build network representation 2017-06-05 18:03:34 -07:00
batch_softmax_loss.py add support for weight in batch_softmax_loss 2017-06-21 10:32:15 -07:00
build_index.py Return top K classes 2017-07-13 00:20:00 -07:00
concat.py dot product using matmul 2017-07-17 23:20:37 -07:00
dot_product.py Dict fixes/improvements and unittest targets for Python 3 in caffe2 core 2017-06-29 17:05:41 -07:00
dropout.py Adding Dropout Layer to SparseNN Model and Flow 2017-06-20 15:46:55 -07:00
fc.py
fc_without_bias.py
feature_sparse_to_dense.py Rename SparseToDense layer 2017-06-09 12:48:27 -07:00
functional.py make functional layer return scalar if only one output 2017-07-12 11:34:31 -07:00
gather_record.py
last_n_window_collector.py
layers.py IndexHash 2017-07-07 23:06:11 -07:00
pairwise_dot_product.py BatchGatherOp 2017-07-27 10:17:42 -07:00
position_weighted.py add truncation for sparse feature 2017-06-13 10:46:53 -07:00
random_fourier_features.py Simplifying Random Fourier Features and layer test 2017-07-11 00:40:53 -07:00
sampling_train.py Modify samplingTrain layer to take more general inputs 2017-07-08 22:19:55 -07:00
sampling_trainable_mixin.py
select_record_by_context.py
semi_random_features.py Modify arc cosine feature map and semi random layers to initialize parameters as global constants 2017-07-26 16:37:00 -07:00
sparse_feature_hash.py IndexHash 2017-07-07 23:06:11 -07:00
sparse_lookup.py make SparseLookup support None pooling 2017-07-18 16:39:55 -07:00
split.py
tags.py offline_all_gpu_experiment 2017-06-27 23:09:54 -07:00
uniform_sampling.py