11 Commits

Author	SHA1	Message	Date
Adam Paszke	23ab9d481a	Add Module._all_buffers	2017-06-12 21:58:38 -04:00
Sam Gross	65b66264d4	Improve broadcast/reduce performance by coalescing tensors	2017-03-06 12:47:53 -08:00
Adam Paszke	d6fa3b3fd5	Deprecate nn.Container in favor of nn.Module	2017-01-16 19:07:37 -05:00
Adam Paszke	8d60e39fdc	Rename torch.nn.functions to torch.nn._functions	2016-12-30 23:02:57 +01:00
Adam Paszke	aa8916e7c6	Don't unpack single element tuples returned by functions	2016-11-23 18:48:41 +01:00
Adam Paszke	80a827d3da	Fix data_parallel bugs	2016-11-23 18:48:41 +01:00
Sam Gross	15377ac391	Copy Module._buffers in nn.parallel.replicate (#180)	2016-10-31 12:12:29 -04:00
Sam Gross	f4ebc65a12	Add Module.modules() and Module.children() (#90) ... modules(): returns an iterator over all modules in the network children(): returns an iterator over immediate children Also fix __getitem__ in Sequential	2016-10-01 21:18:53 -04:00
Sam Gross	e8a5f00866	Auto GPU for CUNN (#71)	2016-09-30 14:04:53 -04:00
Soumith Chintala	412019dbe4	fixing CPU builds by making cuda imports optional	2016-09-28 11:56:18 -04:00
Adam Paszke	3eac7164f4	Add data parallel functions to nn	2016-09-27 15:45:45 -07:00