pytorch/caffe2/python/helpers
Ilia Cherniavskii 79ac146808 Add if and while ops to brew
Summary:
Adding if and while control ops to brew, also adding unit tests
Note: unlike net_builder where we can figure which blobs are external and which ones are local to subnets, here in brew we need to use external_blobs param explicitly to point at external blobls

Reviewed By: harouwu

Differential Revision: D6440508

fbshipit-source-id: c920f0af84b77ccb2d8462ffc7567bb1908c844a
2017-12-05 17:33:34 -08:00
..
algebra.py Re-license to Apache 2017-09-28 16:22:00 -07:00
arg_scope.py Re-license to Apache 2017-09-28 16:22:00 -07:00
array_helpers.py Re-license to Apache 2017-09-28 16:22:00 -07:00
control_ops.py Add if and while ops to brew 2017-12-05 17:33:34 -08:00
conv.py Re-license to Apache 2017-09-28 16:22:00 -07:00
dropout.py Disable cudnn dropout 2017-10-04 17:24:09 -07:00
elementwise_linear.py Re-license to Apache 2017-09-28 16:22:00 -07:00
fc.py Added FP16 compute support to FC Op 2017-10-30 17:03:51 -07:00
nonlinearity.py Re-license to Apache 2017-09-28 16:22:00 -07:00
normalization.py update transformer code for layer_norm() API change 2017-12-01 20:19:31 -08:00
pooling.py Re-license to Apache 2017-09-28 16:22:00 -07:00
tools.py Add missing is_test argument in ImageInput ops 2017-10-10 10:03:13 -07:00
train.py Re-license to Apache 2017-09-28 16:22:00 -07:00