Commit Graph

505 Commits

Author SHA1 Message Date
Andrew Tulloch 69ad6aed82 [NNPACK] Add check for NNPACK being available (nnp_initialize() succeeding) (#2119)
This fixes issues with failing tests on PowerPC.
2018-11-15 14:01:36 -08:00
Tianqi Chen b252160451 [RELAY][PASS] Bind, FoldConstant (#2100) 2018-11-14 08:56:40 -08:00
Yao Wang 313e1d99a8 [Relay][OP]NMS (#1929) 2018-11-13 14:57:16 -08:00
Tianqi Chen 1f2c815671 [RELAY][OP] strided_slice (#2094) 2018-11-13 13:32:38 -08:00
Tianqi Chen 4369b7f6a8 [RELAY][PASS] General OpFusion. (#2090) 2018-11-13 13:31:45 -08:00
Wuwei Lin 5c410c4c31 [TOPI][CUDA] int8 group conv2d (#2075) 2018-11-10 08:52:22 -08:00
Andrew Tulloch fc83c7f26d [TVM] [NNPACK] Modernize and improve NNPACK bindings (#2084) 2018-11-09 18:00:26 -08:00
Tianqi Chen 9f441d817c [RELAY] CompileEngine update, nn conv2d, fix dense, pool. (#2082) 2018-11-09 14:25:53 -08:00
Wuwei Lin c294dd5d73 Allow to use negative index of array in python (#2069)
* Allow to use negative index of array in python

* Support negative index in array slice

* Print index and array size in IndexError

* Fix style
2018-11-06 10:42:21 -08:00
Tianqi Chen c91ded3267 [RELAY][BACKEND] CompileEngine refactor. (#2059) 2018-11-04 22:02:27 -08:00
Jared Roesch 4e77eeb22f [RELAY][RUNTIME] Add compute and schedule attributes for all ops in relay/op/tensor.py (#2050) 2018-11-04 17:46:21 -08:00
Jared Roesch ead3ac6c23 Rename relay::Environment to relay::Module (#2054) 2018-11-02 14:27:34 -07:00
Jared Roesch c48812dd38 [RELAY][RUNTIME] Refactor interpreter and graph_runtime into consistent interface. (#2042) 2018-10-31 19:41:47 -07:00
Wuwei Lin 2005f85237 [TOPI] Add dilation argument to conv2d and depthwise_conv2d (#1970) 2018-10-31 15:12:44 -07:00
eqy 7a3e389bd5 [RELAY] MobileNet (#1997) 2018-10-31 10:51:19 -07:00
ziheng dd9d76acf4 [RELAY/PASS] Simplify inference. (#2033) 2018-10-31 09:51:54 -07:00
Lianmin Zheng f2d7787e57 [AUTOTVM] Misc fix to document and style (#2035) 2018-10-30 19:20:53 -07:00
Jared Roesch 10ea05e645 [RELAY][RUNTIME] Add Relay interpreter and compiler for TVM runtime system. (#1954) 2018-10-30 15:29:36 -07:00
Tianqi Chen d5103bbcdf [RELAY][PASS] FoldScaleAxis Backward (#2024) 2018-10-29 19:43:37 -07:00
Yizhi Liu b9e8826f82 Refine porting x86 NCHWc conv to AutoTVM (#1993) 2018-10-29 14:36:44 -07:00
Siju e286e6378b [RELAY]prelu op support (#2016) 2018-10-29 14:15:37 -07:00
Josh Pollock 2fb1cc6eb5 [Relay] DCGAN port (#2010) 2018-10-29 14:10:06 -07:00
Josh Pollock ed1718b631 [Relay] DQN Port (#2009) 2018-10-28 20:56:58 -07:00
xqdan 32f158e6c2 [intrin]support fmod for cuda (#1964) 2018-10-28 18:37:05 -07:00
Tianqi Chen 05ea6018e5 [RELAY][PASS] FoldScaleAxis Forward (#2020)
* [RELAY][PASS] FoldScaleAxis Forward

* Introduce helper function type_as

* Update per review comment

* Fix according to comments
2018-10-28 15:59:26 -07:00
Siju 62a94c7683 [RELAY]reshape_like (#1950) 2018-10-26 20:14:15 -07:00
Siva e6e9b37170 [RELAY][OP] Split (#1876) 2018-10-26 12:15:16 -07:00
eqy 44df04a1ef fix typo in resnet definition (#1995) 2018-10-25 18:30:23 -07:00
Jared Roesch 0f7aa30b7b [RELAY] Add structural hashing for Relay (#1977) 2018-10-25 14:33:21 -07:00
Tianqi Chen c76fce9fd9 [RELAY] BiasAdd, MLP, Resnet testing (#1969)
* [RELAY] BiasAdd, MLP, Resnet testing

* fix review comments
2018-10-24 13:51:12 -07:00
雾雨魔理沙 399b39f175 fix pydoc format (#1975) 2018-10-24 11:44:18 -07:00
Steven S. Lyubomirsky 3a1bb8c7d4 [Relay] Serialization round-trip tests (#1968) 2018-10-23 18:38:41 -07:00
Lianmin Zheng 390acc52a1 [AUTOTVM] Fix measurement for CPU (#1956) 2018-10-22 21:55:31 -07:00
Yao Wang e61659a959 Add tophub for x86 (#1955) 2018-10-22 21:55:16 -07:00
雾雨魔理沙 c2b3615477 [Relay][Op]BroadcastToLike CollapseSumLike (#1886) 2018-10-22 09:31:59 -07:00
雾雨魔理沙 c51268c3ee [Relay] fix doc in ty.py (#1949) 2018-10-21 17:33:29 -07:00
雾雨魔理沙 af1fcf9743 [Relay] fix format in ty.py (#1948) 2018-10-21 10:30:04 -07:00
Siju 02a8be10a0 [RELAY]Reduce ops sum/max/min/mean/prod (#1927) 2018-10-21 09:46:32 -07:00
Sergey Mironov 39c8bc2a3d [TOPI] Specify non-zero absolute tolerance in tests (#1925) 2018-10-20 22:06:45 -07:00
Alexander Pivovarov a22a057fb5 Fix typo in module.py line 90 (#1947) 2018-10-20 21:53:41 -07:00
Tianqi Chen 8876eac8af [RELAY] IR builder stablize refactor, clean pass (#1934) 2018-10-19 21:56:09 -07:00
Tianqi Chen 9d44279eb9 [RELAY][Refactor] TextPrinter, move ret_type after body in Function. (#1918) 2018-10-18 13:06:48 -07:00
Siju 4001569e5e [RELAY][OP]Reduction operator framework, argmax, argmin (#1865) 2018-10-18 08:53:51 -07:00
Yao Wang dc996e451a AutoTVM x86 (#1772)
* AutoTVM for x86 conv2d

* Add ApplyGraphBest dispatch context

* Fix tutorial

* Fix conv2d

* Improve tutorial

* Fix default schedule

* Fix 1x1 default schedule loading

* Fix workload type

* Change gridsearch to random

* Add reference to autotvm arm

* Merge conv2d common and 1x1 decl

* Fix lint

* Minor fix
2018-10-17 18:20:40 -07:00
Siju d5d194494d [RELAY]Ops Dense, leaky_relu (#1828) 2018-10-16 18:27:41 -07:00
Steven S. Lyubomirsky b4946e770c [Relay][Op] Dropout and batch_norm (#1870) 2018-10-15 17:10:20 -07:00
Yao Wang b64f3f1cfb [Relay][OP] MultiboxPrior (#1882)
* Relay MultiboxPrior Operator

* Fix lint

* Fix build

* Add test for default args
2018-10-15 12:16:51 -07:00
雾雨魔理沙 201cfdc59a [Relay] [Op] Squeeze (#1858) 2018-10-15 09:46:21 -07:00
Siju 47b8c36dcf [RUNTIME][DEBUG]Support remote debugging (#1866) 2018-10-14 18:13:52 -07:00
Tianqi Chen 0b4cc05050 [RELAY][IR] Move type_annotation to Var, remove Param (#1900) 2018-10-14 18:08:52 -07:00