mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
Summary: This is going to unblock Nvidia in their work on adding fp16 support to Caffe2. I discussed this with kennyhorror before to make sure this fits into his work on parameter sharing. Reviewed By: kennyhorror Differential Revision: D5127797 fbshipit-source-id: 4db155d320b1862570c23b77c4252bdacbf2296f |
||
|---|---|---|
| .. | ||
| algebra.py | ||
| arg_scope.py | ||
| array_helpers.py | ||
| conv.py | ||
| dropout.py | ||
| fc.py | ||
| nonlinearity.py | ||
| normalization.py | ||
| pooling.py | ||
| tools.py | ||
| train.py | ||