pytorch/test
Xiang Gao 6fc75eadf0 Add CELU activation to pytorch (#8551)
Summary:
Also fuse input scale multiplication into ELU

Paper:
https://arxiv.org/pdf/1704.07483.pdf
Pull Request resolved: https://github.com/pytorch/pytorch/pull/8551

Differential Revision: D9088477

Pulled By: SsnL

fbshipit-source-id: 877771bee251b27154058f2b67d747c9812c696b
2018-08-01 07:54:44 -07:00
..
bottleneck Implement torch.util.bottleneck (#5216) 2018-03-23 17:27:35 -04:00
cpp/api Add CELU activation to pytorch (#8551) 2018-08-01 07:54:44 -07:00
cpp_extensions Creates CUDAContext (#9435) 2018-07-20 12:56:15 -07:00
data Merge Variable and Tensor classes (#5225) 2018-02-23 18:03:31 -05:00
error_messages
expect Slightly relax the constraints on argument and return types to script functions (#9969) 2018-07-31 14:25:29 -07:00
ffi/src
onnx Add CELU activation to pytorch (#8551) 2018-08-01 07:54:44 -07:00
optim
common.py Re-enable empty n-dimensional empty tensor and fix parallel CPU on empty tensors (#10077) 2018-07-31 16:43:45 -07:00
common_cuda.py Add memory leak check in CUDA tests (#7270) 2018-05-31 15:09:54 -04:00
common_nn.py Add CELU activation to pytorch (#8551) 2018-08-01 07:54:44 -07:00
run_test.py Create initial Python bindings for c10d (#8119) 2018-06-08 12:59:51 -07:00
test_autograd.py Re-enable empty n-dimensional empty tensor and fix parallel CPU on empty tensors (#10077) 2018-07-31 16:43:45 -07:00
test_c10d.py Test case for c10d DDP (#9670) 2018-07-24 14:10:24 -07:00
test_cpp_extensions.py Export tensor descriptor (#8313) 2018-06-20 22:32:50 -07:00
test_cuda.py Implement add, sub, mul, div using TensorIterator (#8919) 2018-07-27 14:43:24 -07:00
test_dataloader.py Revert "Fix dataloader hang when it is not completely iterated (#9655)" (#9804) 2018-07-25 10:10:30 -07:00
test_distributed.py Build system changes (#8627) 2018-06-20 17:45:26 -04:00
test_distributed_trap.py fbshipit-source-id: ba600fcd2b5cefc7621357bdeb05e24cea02e5af 2018-06-27 04:50:56 -07:00
test_distributions.py Introduce __array_priority__ on torch.Tensor (#9651) 2018-07-30 14:39:43 -07:00
test_indexing.py Re-enable empty n-dimensional empty tensor and fix parallel CPU on empty tensors (#10077) 2018-07-31 16:43:45 -07:00
test_jit.py Slightly relax the constraints on argument and return types to script functions (#9969) 2018-07-31 14:25:29 -07:00
test_legacy_nn.py Add CTC loss (#9628) 2018-07-31 11:09:48 -07:00
test_multiprocessing.py Reenable multiprocessing preserve sharing tests on ASAN. (#9498) 2018-07-17 11:10:21 -07:00
test_nccl.py Add memory leak check in CUDA tests (#7270) 2018-05-31 15:09:54 -04:00
test_nn.py Add CELU activation to pytorch (#8551) 2018-08-01 07:54:44 -07:00
test_optim.py Changed serialization mechanism of LambdaLR scheduler (#9927) 2018-07-31 19:39:06 -07:00
test_sparse.py Test nn.Module on non-contiguous inputs (#9114) 2018-07-05 21:09:34 -07:00
test_torch.py fix test_load_error_msg failure (Network is unreachable) (#10021) 2018-08-01 00:24:01 -07:00
test_utils.py Nuke TestCollectEnv (#9459) 2018-07-16 13:10:28 -07:00