Commit Graph

23 Commits

Author SHA1 Message Date
Dun Liang 86a3feeaab polish pool interface 2021-08-04 19:28:54 +08:00
Dun Liang 2db62c17a6 polish pool interface 2021-07-05 21:20:10 +08:00
Dun Liang 8cab7d2a3e support different stride and kernel for unpool 2021-07-05 17:11:43 +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 bb2d187dd3 add max_pool with index and max_unpool2d 2021-04-16 16:57:52 +08:00
Dun Liang fba55c7e31 polish pool interface 2021-04-15 14:33:51 +08:00
zhouwy19 99b7fbeec1 change pool inference 2021-03-23 20:32:07 +08:00
jwzxgy2007 9f8da7507c fix bug 2021-02-05 16:45:16 +08:00
Dun Liang 3f858b76e1 update 2020 -> 2021 2021-01-07 16:25:00 +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
jwzxgy2007 81d8c528db add pool3d 2020-12-16 22:32:36 +08:00
Dun Liang 79579b4a30 merge loop var optimize 2020-11-25 22:15:41 +08:00
Dun Liang d798456fee fix ci 2020-05-28 23:20:45 +08:00
Dun Liang a6dab87634 add count_include_pad for pad 2020-05-27 22:19:16 +08:00
Dun Liang 93ca5e9525 add cuda limit.h 2020-05-27 17:28:34 +08:00
Dun Liang 1e867a5b7b fix avg pool compare with pytorch 2020-05-27 16:59:00 +08:00
Dun Liang ab533a5f45 code op增加更多特性 :
* 可以有多个输出
* 允许输出动态大小的var
* code op内部可以写注释了
* code op可以使用别名了,通过@alias为inputa和outputs增加别名
2020-04-26 20:46:19 +08:00
zwy 6e08781df4 models & converter & flatten/view/permute/adapool 2020-04-19 22:01:32 +08:00
guoye 9d9895190e set to 0 2020-03-30 13:10:57 +08:00
guoye 0625957143 pool 2020-03-27 18:01:59 +08:00