Commit Graph

132 Commits

Author SHA1 Message Date
Dun Liang 88483fedbc add layernorm3d 2021-06-04 14:07:07 +08:00
Dun Liang 6eafa4dd66 Merge branch 'master' of https://github.com/jwzxgy2007/jittor 2021-06-04 14:01:11 +08:00
Dun Liang 2efd7be357 add UpsamplingBilinear2d, AdaptiveMaxPool2d 2021-05-11 16:29:59 +08:00
Dun Liang 862d564d85 add distribution for rl 2021-04-29 23:04:02 +08:00
li-xl b75b6def89 add rnncell and grucell 2021-04-29 16:08:05 +08:00
lzhengning b0669c11c0 feats: polish rnnbase 2021-04-29 15:05:39 +08:00
lzhengning 026dfb8fa2 feat: add rnn, gru 2021-04-29 14:58:24 +08:00
lzhengning 18923fcf3d feat: bidirectional 2021-04-29 13:32:18 +08:00
lzhengning 5f6c170121 feat: add lstm 2021-04-28 15:11:08 +08:00
li-xl 8f7c997afb update upsample 2021-03-12 15:34:34 +08:00
li-xl 9b67168f18 update zeropad 2021-03-11 16:08:52 +08:00
Dun Liang 36e7acfc1f polish auto_parallel and upsample 2021-03-11 16:05:34 +08:00
Dun Liang 8616c01a74 polish conv1d bias 2021-03-10 15:25:55 +08:00
Dun Liang 40ce9e605a polish conv1d weight 2021-03-10 13:35:06 +08:00
Exusial 0d7f0db1b9
add unfold,fold,bicubic,qr,eye (#180)
* fix bicubic,add fold.

* add eye.

* add test for qr,bicubic,fold,unfold.

* fix bicubic and fold to code_op ver.add grad test.

* add docs.update pinv to support (..,M,N) shape

* edit maintainer and testfunc's name.

* fix nn

* fix nn

Co-authored-by: Exusial <2247838039@qq.com>
Co-authored-by: Gword <471184555@qq.com>
2021-03-02 16:38:28 +08:00
Dun Liang 73aa382627 fix densenet vary shape input 2021-02-25 11:39:55 +08:00
Dun Liang aefe719770 add stride padding(h,w) support and ctrl-c quick exit 2021-02-20 16:34:24 +08:00
Dun Liang 3f858b76e1 update 2020 -> 2021 2021-01-07 16:25:00 +08:00
Dun Liang 0a5a7f5ea2 update Softplus Module 2020-12-29 10:49:51 +08:00
Dun Liang 2c939cb301 change header 2020-12-25 16:08:31 +08:00
zhouwy19 49ac9be019 add more pool & random method 2020-12-25 13:34:34 +08:00
Dun Liang 1b6970f7dc add requires_grad 2020-12-21 22:41:45 +08:00
li-xl 1f8ea20f01 add elu 2020-12-16 20:03:24 +08:00
jwzxgy2007 b026c3b702 add conv3d 2020-12-13 21:06:34 +08:00
Dun Liang c8252ac7fb add conv transpose function 2020-12-13 19:03:43 +08:00
Dun Liang c80e0168c0 Merge branch 'develop' of https://github.com/Jittor/jittor 2020-12-07 22:49:47 +08:00
Dun Liang 514b60dea8 Merge branch 'master' of https://github.com/Jittor/jittor 2020-12-07 22:49:33 +08:00
cxjyxx_me 895f6ecee2 depthwise_conv 2020-12-07 22:43:11 +08:00
cxjyxx_me 14324f364b mrege 2020-12-02 16:57:36 +08:00
Jittor 260a052db9
Update nn.py
fix a simple bug
2020-11-28 20:27:07 +08:00
li-xl 3c7d59ea16 fix Layernorm bugs 2020-11-27 15:19:31 +08:00
Xiang-Li Li 5dfef0af13
Merge branch 'develop' into lxl 2020-11-27 15:11:36 +08:00
Zheng-Ning Liu d1975d9d7e
Add aliases in nn.py 2020-11-27 14:58:30 +08:00
Dun Liang 79579b4a30 merge loop var optimize 2020-11-25 22:15:41 +08:00
li-xl 13941112e1 Merge branch 'master' of https://github.com/jittor/jittor into lxl 2020-11-25 19:23:28 +08:00
Zheng-Ning Liu 798018880e
Update BatchNorm1d
Change default value of affine in BatchNorm1d to be True.
2020-11-25 13:30:21 +08:00
cxjyxx_me 97a99b6a24 depthwise_conv 2020-11-25 11:13:47 +08:00
li-xl 1045f04c4b add misc and bcelogits with pos_weight 2020-11-10 11:34:39 +08:00
li-xl 32af287461 Merge branch 'master' of https://github.com/jittor/jittor 2020-11-08 16:04:16 +08:00
li-xl 98545bede4 add softplus 2020-11-08 16:04:12 +08:00
Dun Liang 805f48369d better trace_py_var 2020-11-06 14:41:38 +08:00
li-xl cb52493db9 Merge branch 'master' of https://github.com/jittor/jittor 2020-11-03 17:06:20 +08:00
li-xl 06eb505eea add bmm_transpose and pad and merge_loop_flags 2020-11-03 17:04:25 +08:00
zhouwy19 b742939605 update sign 2020-10-28 14:07:42 +08:00
zhouwy19 8e93e36f53 add sign & Sequential __iter__ 2020-10-28 11:31:33 +08:00
li-xl aa502deafb optimize where op && fix vary shape infer 2020-10-26 22:03:53 +08:00
li-xl ca7207e027 Merge branch 'master' of https://github.com/jittor/jittor 2020-10-21 11:37:06 +08:00
li-xl bd506460be remove matmul print 2020-10-21 11:36:59 +08:00
Dun Liang ee268c7456 add __iter__ in sequential 2020-10-17 15:42:14 +08:00
Dun Liang 8d64d98a35 jit_prepare for array op 2020-10-12 18:01:14 +08:00