pytorch/caffe2/python/layers
Tristan Rice 976e614915 caffe2: add PIPELINE tag (#41482)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/41482

This adds a new tag for use with pipeline parallelism.

Test Plan: CI

Reviewed By: heslami

Differential Revision: D22551487

fbshipit-source-id: 90910f458a9bce68f7ef684773322a49aa24494a
2020-07-24 15:25:14 -07:00
..
__init__.py
adaptive_weight.py
add_bias.py
arc_cosine_feature_map.py
batch_huber_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
bpr_loss.py
bucket_weighted.py
build_index.py
concat.py
constant_weight.py
conv.py
dropout.py
fc.py
fc_with_bootstrap.py
fc_without_bias.py
feature_sparse_to_dense.py [2/n][Compute Meta] support analysis for null flag features 2020-07-20 13:13:45 -07:00
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_dropout_with_replacement.py
sparse_feature_hash.py
sparse_lookup.py
split.py
tags.py caffe2: add PIPELINE tag (#41482) 2020-07-24 15:25:14 -07:00
uniform_sampling.py