Граф коммитов

607 Коммитов

Автор SHA1 Сообщение Дата
Yao Wang ee3c1b09b9 [TOPI]Add where operator (#1416) 2018-07-12 20:04:41 -07:00
Tianqi Chen 81db22c56a [RPC] graduate tvm.contrib.rpc -> tvm.rpc (#1410) 2018-07-09 15:23:41 -07:00
Yizhi Liu 0ec70800dd fix CorrectLayout for softmax & log_softmax (#1401) 2018-07-09 13:19:54 -07:00
Qiao Longfei badcb6302b change line seperator of tensorflow/test_forward from CRLF to LF (#1405) 2018-07-09 09:54:53 -07:00
Pariksheet Pinjari b30ae8ac6d [TOPI][DARKNET]Yolo op added (#1372) 2018-07-08 21:50:52 -07:00
Siju 4e7b548efc [DARKNET FRONTEND]Batchnorm added as part of Dense op for running rnn model for next wo… (#1385) 2018-07-08 21:50:28 -07:00
MORITA Kazutaka 05f4362e8a [NNVM][Keras] allow only tensorflow backend (#1392) 2018-07-08 10:48:26 -07:00
Siva 3a0b757c10 [NNVM][TOP] broadcast versions corresponding to topi: mod, max, min, pow, left_shift, right_shift greater, less, equal, not_equal, greater_equal and less_equal. (#1383) 2018-07-07 18:51:57 -07:00
nhynes 4dd41ddb5b Move SGX enclave signing key to gist (#1393) 2018-07-06 10:17:52 -07:00
Albin Joy 10b7757ab6 [NNVM][TENSORFLOW] Fixed variable ops shape parsing issue (#1381) 2018-07-04 20:45:30 -07:00
Siva 2fa0eca13a [NNVM][ONNX] Slice, Floor, Ceil, Clip and MatMul support for frontend #1297 (#1371) 2018-07-04 11:31:02 -07:00
Tianqi Chen d7c600b80a [TOPI] Update parameter name of conv2d (#1380) 2018-07-04 09:04:03 -07:00
Siva 30f09bea53 [AVG POOL] Asymmetric padding (SAME) support. (#1346) 2018-07-04 09:03:40 -07:00
Dayananda V 40ac206438 add take frontend (#1307) 2018-07-04 09:02:25 -07:00
Siva 4503f77b0a [NNVM][TENSORFLOW] Sigmoid op support #1367 (#1369) 2018-07-03 19:15:02 -07:00
Tatsuya Nishiyama 28acb18432 [NNVM] Move FTVMCompute registration of cast, greter, less to C++. (#1370) 2018-07-03 09:45:11 -07:00
ANSHUMAN TRIPATHY 32a55f8814 Prelu bug fix (#1358) 2018-06-30 07:55:15 -07:00
Tatsuya Nishiyama ab0d1862ab [NNVM] Move FTVMCompute registration of the elementwise operator to c++ (#1351) 2018-06-28 21:28:18 -07:00
Tianqi Chen b154e6b904 [NNVM] Initial mixed precision support of conv2d (#1356) 2018-06-28 10:37:04 -07:00
Pariksheet Pinjari 4fb58115ea Strided_slice added in NNVM (#1318) 2018-06-28 08:05:33 -07:00
Yao Wang 71f88611cd Improve schedule load, add slice_like (#1299) 2018-06-26 18:42:17 -07:00
Tatsuya Nishiyama 50681784f4 [TOPI] Add C++ implementation of elementwise operators (#1306) 2018-06-26 18:40:31 -07:00
nhynes 44d8203fc5 [NNVM][TOPI] Add gradients for broadcast_* ops (#1234) 2018-06-26 18:39:45 -07:00
Siva 2f77a127be [NNVM][ONNX] Squeeze and Unsqueese operators. (#1339) 2018-06-26 11:59:53 -07:00
Siva 113b46ec80 [NNVM][ONNX] Shape operator support (limited/differed) - #1297 (#1333) 2018-06-26 10:15:53 -07:00
Siva 373a8caa79 [NNVM][TENSORFLOW] Mobilenet support. (#1335) 2018-06-26 10:15:24 -07:00
Wenhao Hu c3df7726ac support t attr in onnx (#1300) 2018-06-26 09:03:52 -07:00
nhynes 59a8d09936 [NNVM][TOPI] Add FTVMCompute for matmul (#1239) 2018-06-25 20:41:01 -07:00
Siva bb7df695cd [NNVM][CONVOLUTION] Group convolution generalization for NHWC (#1232) 2018-06-24 09:09:09 -07:00
rajh619 5b33e7b898 Windows VS2015 Compile error - 'conversion from 'double' to 'float' requires a narrowing conversion' (#1329) 2018-06-24 09:05:10 -07:00
Tianqi Chen bdd4bab02d [NNVM] Introduce const shift ops (#1325) 2018-06-23 18:57:18 -07:00
Tatsuya Nishiyama f216b25e01 [FRONTEND][MXNET] Add expand_dims supoort (#1317)
* [FRONTEND][MXNET] Add expand_dims supoort

* fix lint
2018-06-23 08:57:27 -07:00
Tianqi Chen db4be63c51 [TOPI] Numpy consistency: always broadcast binary op. (#1321) 2018-06-22 16:49:07 -07:00
Pariksheet Pinjari fb88b74e4a CPP implementation of L2Norm and LRN ops (#1157) 2018-06-22 10:06:40 -07:00
Junru Shao 464c8c2667 Fix a bug in Symbol::Compose when using subgraphs as input (#1314) 2018-06-21 19:25:55 -07:00
Thierry Moreau baa04599da [CONTRIB] TVM download utility based on urllib2/urlib.request (#1313)
moving nnvm/testing/download.py to python/tvm/contrib/download.py to be used as a general TVM download utility
2018-06-21 15:39:43 -07:00
Tianqi Chen 61370e4b6c [MATH][TOPI][NNVM] introduce trunc, round (#1310) 2018-06-21 13:23:54 -07:00
Tatsuya Nishiyama 80e4bc029c [FRONTEND][MXNET] Add squeeze_axis support to split operator (#1288) 2018-06-20 09:58:07 -07:00
Lianmin Zheng 42b189cbc0 [NNVM][TESTING] Add two testing symbols: dqn and dcgan (#1294) 2018-06-17 12:38:23 -07:00
Yao Wang c418e916bc SSD support in NNVM (#1214) 2018-06-14 15:22:08 -07:00
Siva 76fa3ca4f6 [TOPI][IMAGE][RESIZE] Bilinear interpolation for resize and upsampling. (#1181) 2018-06-14 08:53:49 -07:00
Tianqi Chen fed6298b7b [BUILD] Add clang to build matrix, -Werror (#1273) 2018-06-13 10:52:49 -07:00
Siva a81ebd9040 [NNVM][FRONTEND] Tensorflow frontend support (#1188) 2018-06-12 22:37:35 -07:00
Da Zheng 558cf098c0 add support for subgraphs. (#1221)
* add support for subgraphs.

* fix.

* fix.

* Fix compilation error

* Fix compilation error

* add comments.

* update comments.

* Sanity check on subgraphs when creating IndexedGraph

* avoid the overhead of sanity check.

* Stop using non-recursive DFS

* Trigger CI

* trigger CI
2018-06-12 15:46:48 -07:00
Tatsuya Nishiyama 669b44c174 Fix the gemm conversion in onnx frontend (#1241) 2018-06-11 16:43:05 -07:00
Tianqi Chen 552fa71b74 [BUILD] Upgrade build system to default python3 (#1260) 2018-06-11 16:36:52 -07:00
Haibin Lin cd39697e11 Update layout.h (#1255) 2018-06-10 20:49:19 -07:00
abergeron 4be6a30788 Unite cmake builds (#1248) 2018-06-08 10:23:34 -07:00
Tatsuya Nishiyama 5a15664ebe Add PReLU support to mxnet frontend (#1249) 2018-06-07 21:32:52 -07:00
Pariksheet Pinjari 31223fa248 [NNVM]Activations support added in Keras Frontend (#1210)
* [NNVM]Activations support added in Keras Frontend

* Helper for ELU added

* All activations test cases clubbed to one
2018-06-07 08:44:57 -07:00
MORITA Kazutaka 34630e8147 [NNVM][FRONTEND][Keras] Support for reusing layers (#1192) 2018-06-06 08:44:52 -07:00
larrywyang a9313787ab [WIP] [NNVM] Fix softmax gradient (#1201)
[NNVM] Fix softmax gradient
2018-06-05 13:51:38 -07:00
Pariksheet Pinjari ce34ae16f9 [NNVM] Elu support added in darknet frontend (#1199) 2018-06-04 12:12:11 -07:00
Pariksheet Pinjari 9c299a909a leaky_relu bug fix (#1218) 2018-06-01 13:09:53 -07:00
Siva 59c8bc9652 [LINT] Expected, hence disable for a while. (#1212)
************* Module nnvm.top.nn
E:141, 8: Assigning to function call which doesn't return (assignment-from-no-return)
2018-05-31 10:02:35 -07:00
tqchen 52f045171f [RUNTIME] update to make runtime copy type aware 2018-05-30 22:27:44 -07:00
tqchen 42608ddafb [IO] Support cross-endian 2018-05-30 22:27:44 -07:00
tqchen 55d13d47fd Homepage URL to tvm.ai 2018-05-29 16:24:42 -07:00
tqchen 5195679c5c [DOCS] Improve docs naming, fix docs warnings 2018-05-29 08:47:00 -07:00
tqchen bb2b86201c [DOCS] documentation merge 2018-05-29 08:47:00 -07:00
Pariksheet Pinjari 891b4e06ce Flip operator (#505) 2018-05-29 08:47:00 -07:00
Ehsan M. Kermani 31edf3f7eb Expose clip to frontend mxnet (#512) 2018-05-29 08:47:00 -07:00
Siva 90c1157b5f Consider version too #508 (#514) 2018-05-29 08:47:00 -07:00
MORITA Kazutaka a9d2149711 [FRONTEND][Keras] Add test for MobileNet (#513) 2018-05-29 08:47:00 -07:00
MORITA Kazutaka c68945c582 [FRONTEND][Keras] fix reshape (#493) 2018-05-29 08:47:00 -07:00
Tao Lv 39cc9c1265 fix doc (#510) 2018-05-29 08:47:00 -07:00
Siva 510cd5ec81 Squeeze bug fix. (#506) 2018-05-29 08:47:00 -07:00
Tatsuya Nishiyama 69d5fcab74 Add count_include_pad support (#498)
* update tvm submodule

* Add count_include_pad support to avg_pool
2018-05-29 08:47:00 -07:00
MORITA Kazutaka fd6ad2742b [FRONTEND][Keras] Fix softmax axis (#503) 2018-05-29 08:47:00 -07:00
Siva 8534db36b1 Increase depfiles lookup. (#509)
Not able link build/src/top/tensor/*.d.
       Hence don't compile nnvm for a change in tvm/topi headers.
2018-05-29 08:47:00 -07:00
Siva c7474c005a Cleanup of '-Wsign-compare' warnnigs. (#504) 2018-05-29 08:47:00 -07:00
Tang, Cheng 71ffc2b6ee don't rename onnx input tensors (#491) 2018-05-29 08:47:00 -07:00
Pariksheet Pinjari 9f6e0c596c Fixed issue #483, removing enum dependancy (#485) 2018-05-29 08:47:00 -07:00
Siva a349f5ec4f Better to check Infer result with topi results at build time instead of leaving to a runtime error. (#476) 2018-05-29 08:47:00 -07:00
Tang, Cheng 3c3eb3b367 add use_mt option, default to use md as mt won't work with current c++ api (#484) 2018-05-29 08:47:00 -07:00
Yizhi Liu 7f5d22d7f6 enable AlterOpLayout to keep OP unchanged (#471) 2018-05-29 08:47:00 -07:00
MORITA Kazutaka be968fef00 [Keras] ReLU6 support (#481) 2018-05-29 08:47:00 -07:00
Tang, Cheng 8db42c4143 add PATH to dll_path on windows (#479) 2018-05-29 08:47:00 -07:00
Tang, Cheng f92daf9238 use uint32_t (#478) 2018-05-29 08:47:00 -07:00
Leyuan Wang 2864fe4e4d dilation fixed for (1, 1) case (#477) 2018-05-29 08:47:00 -07:00
Siva d47682043b fix lint error (#475) 2018-05-29 08:47:00 -07:00
MORITA Kazutaka dd90b19eeb add sanity check to input shape type (#469) 2018-05-29 08:47:00 -07:00
Wenhao Hu 42d81ccaee raise error when cannot get shape in reshape (#467)
* raise error when cannot get shape in reshape

* fix pylint
2018-05-29 08:47:00 -07:00
Yizhi Liu 73775f4532 AlterOpLayout with tvm.target (#463)
* AlterOpLayout with tvm.target

* fix test
2018-05-29 08:47:00 -07:00
Yizhi Liu 343eb82ca2 fix restore layout in AlterOpLayout (#460)
* fix restore layout in AlterOpLayout

* lint test case
2018-05-29 08:47:00 -07:00
Yizhi Liu 1e4bb2f887 allow missing FCorrectLayout (#457)
* allow missing FCorrectLayout

* misunderstood OpMap[], fix
2018-05-29 08:47:00 -07:00
Liangfu Chen 1b5877f28d [DOCS] improve document of symbol; (#456) 2018-05-29 08:47:00 -07:00
Tatsuya Nishiyama 11ebd8d303 support dilation in conv2d (#439) 2018-05-29 08:47:00 -07:00
Yizhi Liu d5744844c1 Rename FInferLayout -> FCorrectLayout (#453)
* rename FInferLayout -> FCorrectLayout

* correct stupid IDE

* update submodule tvm
2018-05-29 08:47:00 -07:00
Yizhi Liu 9f8fcfc9bc General Layout Support (#447) 2018-05-29 08:47:00 -07:00
Pariksheet Pinjari fc7e9cd206 Some minor documentation issues rectified (#450) 2018-05-29 08:47:00 -07:00
MORITA Kazutaka 386fbb3f43 Fix license URL (#451) 2018-05-29 08:47:00 -07:00
Pariksheet Pinjari a82656d497 Support imagescalar operator for onnx (#448) 2018-05-29 08:47:00 -07:00
Yizhi Liu 435201ee12 fix mxnet model import (#449) 2018-05-29 08:47:00 -07:00
Pariksheet Pinjari 4b765c511d [OP] PReLU Support (#394) 2018-05-29 08:47:00 -07:00
Pariksheet Pinjari 079599c95d Issue fix for tiny yolo (#436)
* Issue fix for tiny yolo

There was an issue in maxpool frontend for tiny-yolo. It is fixed in PR. 
https://github.com/dmlc/nnvm/issues/431

* updated review comment, python 3 compatabilty

* Update darknet.py

* Updated review comment
2018-05-29 08:47:00 -07:00
Tianqi Chen d5b9f35487 Update TVM to latest (#432)
* Update TVM to latest

* remove darknet from testing due to cffi
2018-05-29 08:47:00 -07:00
Pariksheet Pinjari 7ec898d55e [FRONTEND] DarkNet Yolo2 Frontend Support (#377) 2018-05-29 08:47:00 -07:00
Haibin Lin 2e836ca768 declare type name for optional<TShape> (#429)
* declare type for optional tshape

* add doc

* move code to another place
2018-05-29 08:47:00 -07:00
Tianqi Chen a53d8d0172 [PASS] Enhance scale fold axis (#424) 2018-05-29 08:47:00 -07:00
Tianqi Chen 89c124bc89 Update tvm, fix lint due to pylint update (#423) 2018-05-29 08:47:00 -07:00
Wenhao Hu e722dbcbf7 Onnx opset support (#416) 2018-05-29 08:47:00 -07:00
Yao Wang f4789db696 Add optimizer (#334) 2018-05-29 08:47:00 -07:00
Wenhao Hu bd40bcd1a2 use caffe2.python.onnx.backend instead of onnx-caffe2 (#418) 2018-05-29 08:47:00 -07:00
Wenhao Hu ec5c46519b fix onnx frontend softplus bug (#413) 2018-05-29 08:47:00 -07:00
Yuwei Hu 8c5b4909f4 [Keras] fix dropout bug (#399) 2018-05-29 08:47:00 -07:00
Yan Huang b7b74228ff [FIX] several bugs found when using NNVM (#391) 2018-05-29 08:47:00 -07:00
Pariksheet Pinjari a523311db0 Documentation errors updated (#392) 2018-05-29 08:47:00 -07:00
Tianqi Chen 76ca335215 fix tutorial build (#390) 2018-05-29 08:47:00 -07:00
Tianqi Chen 6a7d1c4567 [OP] Experimental assign op (#389) 2018-05-29 08:47:00 -07:00
nhynes 74a3f74208 Add end-to-end SGX ResNet inference example (#388) 2018-05-29 08:47:00 -07:00
Atsushi Nukariya 3f22f1900a [DOCS] add a solution to installation error because of the old submodule (#384)
* [DOCS] add a solution to installation error

* Fix the command for updating submodules
2018-05-29 08:47:00 -07:00
Siva 33ac203761 Sum operator for ONNX. (#386) 2018-05-29 08:47:00 -07:00
Zhixun Tan 4794f3d6ec WebGL end-to-end example (#369) 2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 9ba95ec47a [FRONTEND] fix same var used in single op (#383)
* fix same var used in single op

* revert to older version
2018-05-29 08:47:00 -07:00
Tianqi Chen a3a9dbeb9a Make compiler more robust (#378) 2018-05-29 08:47:00 -07:00
masahi 860adec806 Initial NHWC layout support (#376)
* initial NHWC layout support

* remove layout param from softmax

* more nhwc support

* fix typo

* add nhwc layout test

* fix lint

* update tvm

* update for c++ topi

* fix lint

* update tvm
2018-05-29 08:47:00 -07:00
Siva 68c039442e [NHWC] InferShape Layout conversion fix. (#372) 2018-05-29 08:47:00 -07:00
Siju Samuel 50c20b76d3 [Compilation Warning Fix]Enum constant in boolean context (#365)
The below compilation warning is fixed.

In file included from /mnt/D_DRIVE/work/nnvm_8_feb/dmlc-core/include/dmlc/any.h:16:0,
                 from include/nnvm/./base.h:11,
                 from include/nnvm/graph.h:15,
                 from src/compiler/fold_scale_axis.cc:6:
src/compiler/fold_scale_axis.cc: In function ‘nnvm::Graph nnvm::compiler::FoldScaleAxis(nnvm::Graph)’:
src/compiler/fold_scale_axis.cc:155:39: warning: enum constant in boolean context [-Wint-in-bool-context]
     CHECK(kind == kPassTroughFirst || kMulConsumer);
                                       ^
/mnt/D_DRIVE/work/nnvm_8_feb/dmlc-core/include/dmlc/./logging.h:110:9: note: in definition of macro ‘CHECK’
   if (!(x))
2018-05-29 08:47:00 -07:00
Yao Wang f1e0a55abc Move dense compute back to python (#364) 2018-05-29 08:47:00 -07:00
alex-weaver 5884cd01f3 Change TOPI ops to use C++ implementation where applicable (#357)
* Updated TVM version. Implemented fix for nnvm_compiler crash on exit on windows. Changed TOPI ops from using python to using C++ where applicable.

* Fix lint

* Fix lint

* Fix macro

* Fix reshape

* Update TVM to fix test fails
2018-05-29 08:47:00 -07:00
Siju Samuel 3f6423aa2d [Compilation Warning Fix] in matrix_op.cc (#356)
The below compilation warning issue has been fixed.

src/top/tensor/matrix_op.cc:37:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for (int i = 0; i < lshape.ndim() - 1; i++) oshape[i] = lshape[i];
                     ^
src/top/tensor/matrix_op.cc:38:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for (int i = 1; i < rshape.ndim(); i++) oshape[i + lshape.ndim() - 1] = rshap
2018-05-29 08:47:00 -07:00
Siju Samuel ec37028c09 [PYLINT FIX]pylint issue fixes (#348)
Before Fix
************* 
************* Module nnvm._base
R: 21, 0: Disallow trailing comma tuple (trailing-comma-tuple)
R: 27, 0: Disallow trailing comma tuple (trailing-comma-tuple)

-----------------------------------
Your code has been rated at 9.99/

After Fix
************* 
siju@siju-DAEMON:~/D/Community/nnvm$ make lint
pylint python/nnvm --rcfile=/media/siju/DATA/Community/nnvm/tests/lint/pylintrc

-------------------------------------------------------------------
Your code has been rated at 10.00/10 (previous run: 9.99/10, +0.01)
2018-05-29 08:47:00 -07:00
wl3b10s 6be636e1bb image tranform runtime wrong on python (#354)
x = np.transpose(img, (2, 0, 1))[np.newaxis, :]
TypeError: an integer is required
2018-05-29 08:47:00 -07:00
Liangfu Chen 16c28d0818 improved empty set definition (#346)
fix similar problem like https://github.com/apache/incubator-mxnet/pull/4589/files .
2018-05-29 08:47:00 -07:00
ziheng 7023101178 [FIX] Fix target in tutorial (#347) 2018-05-29 08:47:00 -07:00
Siva eb9442e0f2 [Convolution] Error while importing onnx model with weights. (#345)
* [Convolution] Error while imported onnx model has weights.

Fix the use_bias based on the input parameters to use bias (Ex: sqeezenet). Ref #336

* Review corrections.
2018-05-29 08:47:00 -07:00
Yida Wang ffcb2a5ef0 explain the lowering process in nnvm.compiler.build (#339) 2018-05-29 08:47:00 -07:00
Lianmin Zheng 6c19862176 Add CacheItem2Schedule Extension (#338)
* add CacheItem2Schedule extension

* fix lint

* move function position

* make cache item visible to frontend
2018-05-29 08:47:00 -07:00
masahi 2bb4a1e7f0 [TUTORIAL] add extern lib tutorial (#331)
* add extern lib tutorial

* do fix
2018-05-29 08:47:00 -07:00
thefiddler 9339d31f23 Add test for issue #327 (#328)
* Added test for adding down-/up-sampled layers

* Enabled test for adding down-/up-sampled layers

* Normalize whitespace
2018-05-29 08:47:00 -07:00
masahi 82de081cac fix upsampling fuse pattern (#330) 2018-05-29 08:47:00 -07:00
thefiddler 9998f3302f Fix keras frontend elementwise-ops for lists with len>2 (fixes issue #325) (#326)
* Added elementwise-add test

* Fix typo

* Fixed elem-wise ops for lists with len>2
2018-05-29 08:47:00 -07:00
Lianmin Zheng 14de1dec78 [TUTORIAL] add tutorial for mali gpu (#313)
* add tutorial for mali gpu

* update submodule version

* fix doc error

* fix server error

* fix server in test
2018-05-29 08:47:00 -07:00
Siva 963a041973 [NNVM] : Function header corrections. (#324) 2018-05-29 08:47:00 -07:00
Yao Wang 5405127199 Explain how to generate module library in Quick Start tutorial (#323)
* Explain how to generate module library

* Small fix
2018-05-29 08:47:00 -07:00
yuruofeifei f34e1744a2 [GRADIENT] Register more gradient operators (#300)
* Add conv2d max_pool backward op

* Added tests

* Fix testing

* Address comments

* Change dot to matmul

* Address comments

* Break down indicator function

* Make greater, less numpy compatible
2018-05-29 08:47:00 -07:00
Yuwei Hu 4bd92a4a21 [Keras] fix convert_pooling with same pad (#322) 2018-05-29 08:47:00 -07:00
thefiddler 53361fae2a Update frontend for keras 2.1.3 compatibility (#314)
* Keras keeps renaming properties. Update frontend for keras 2.1.3 compatibility

* Add error message when inbound_nodes is not found
2018-05-29 08:47:00 -07:00
Yuwei Hu d626eab87c [CI] turn on keras frontend test (#309)
* [CI] turn on keras frontend test

* fix

* using tensorflow cpu version
2018-05-29 08:47:00 -07:00
guopinglong 2374852e6d make shape inference of BatchNorm layout neutral (#301)
* make shape inference of BatchNorm layout neutral

* refactor to use the axis variable to do BatchNorm shape inference

* refactor to use the axis variable to do BatchNorm shape inference

* add unittest to the axis param for batch norm shape inference
2018-05-29 08:47:00 -07:00
thefiddler 7112dd78f0 Avoid crash when linear activation does not have alpha and beta defined (#306) 2018-05-29 08:47:00 -07:00
Lianmin Zheng 292609d885 remove dtype in model symbol (#310) 2018-05-29 08:47:00 -07:00
thefiddler acb9fd62dd Add support for keras upsampling (#305)
* Add support for keras upsampling

* Fix code formatting

* Fix indentation

* Fix indentation round 2

* Hide unused parameter

* Only enable UpSampling2D since the others are not currently supported by TVM

* Improve error messages and code layout
2018-05-29 08:47:00 -07:00
masahi 6a0fb6efbf Upsampling op support (#298)
* add nnvm upsampling symbol

* add upsampling mxnet frontend

* add doc for upsampling op

* cleanup upsampling test

* minor fix

* use schedule_injective for upsampling

* upgrade tvm
2018-05-29 08:47:00 -07:00
Joshua Z. Zhang d059dbabf0 [WIP] Onnx1.0 (#294)
* add more op for onnx 1.0

* fix syntax

* fix lint

* fix

* update 1.0

* fix

* update model
2018-05-29 08:47:00 -07:00
Hector Li 063cd41293 Add __declspec(dllexport) to class Node and class Op if build on Wind… (#296)
* Add __declspec(dllexport) to class Node and class Op if build on Windows, so that it can be referenced in some test code in MXNet

* correct typo

* reuse the macro in c_api.h

* revert changes on base.h

* one more place to revert

* use enclosing extern c block

* revert a change caused by copy

* remove a space
2018-05-29 08:47:00 -07:00
masahi 3dd7d6467b dump lowered IR when debug logging (#292)
* dump lowered ir when debug log

* avoid calling tvm.lower() twice when not debug
2018-05-29 08:47:00 -07:00
Iblis Lin 8473e9face typo in c_api.h (#290) 2018-05-29 08:47:00 -07:00
Daniel M. Weeks e06d24a68e Add missing cmath header (#287)
f9ed337552a26cc55c7c0fb22cd54839ee13f19c added a call to std::log
without including cmath
2018-05-29 08:47:00 -07:00
Nehal J Wani 38b4ad0a82 Make choice of archiver configurable (#288)
At present, the build system assumes that the Unix archiv utility 'ar'
will always be present in PATH, but this might not be true if a separate
toolchain is being used. This patch makes the build system a bit more
cross-compile capable.
2018-05-29 08:47:00 -07:00
ziheng c1a7b28ba3 [SYMBOL] Enable get index from symbol (#286) 2018-05-29 08:47:00 -07:00
Yao Wang 669e9c15a2 Add gradient graph (#280)
* Add creating gradient symbol

* Fix lint

* Address comments

* Fix typo

* Address comment
2018-05-29 08:47:00 -07:00
Yuwei Hu 492a37c56f [FIX] make keras frontend py3 compatible (#283) 2018-05-29 08:47:00 -07:00
yuruofeifei 999dd1efb9 [GRADIENT] Add backward operator to enable backward graph (#276)
* Update docs

* Add backward operator to enable backward graph

* Fix testing

* Refactor top level1 test code

* Fix format

* Test

* Added zeros ones op

* Register fill_like operator

* Fix unit test
2018-05-29 08:47:00 -07:00
Tianqi Chen e36fb36065 Update deploy_model_on_rasp.py 2018-05-29 08:47:00 -07:00
Yao Wang 275b41d71f Add compile library tutorial (#277)
* Add compile library tutorial

* Clean output

* Refactor with sphinx gallery

* Refactor

* Change title and other minor fixes
2018-05-29 08:47:00 -07:00
Yuwei Hu 083a418436 keras frontend tutorial (#278)
* keras frontend tutorial

* fix
2018-05-29 08:47:00 -07:00
Yuwei Hu e6319f62d0 Keras Frontend (#273)
* vgg16 success

* remove six.PY2, use sys.version_info;
convert_activation() accepts activation type name(str, e.g. 'relu') as input;

* add convert_merge

* fix convert_batchnorm;
improve tests

* fix lint

* add numpy-style pad operator

* deal with asymmetry padding

* resnet50 success

* fix pool_convert; xception passes test

* update tvm

* fix bias error; all tests pass

* use > >, not >>
2018-05-29 08:47:00 -07:00
abergeron 2406757d13 Add support for missing uint types. (#272) 2018-05-29 08:47:00 -07:00
ziheng 02141d4a5e [SYMBOL] Add __iter__ and GetChildren for symbol (#268)
* [SYMBOL] Add __iter__ and GetChildren for symbol

* [SYMBOL] Fix lint
2018-05-29 08:47:00 -07:00
Chris Olivier a0dc865529 API call to get symbol output count (#270)
* Symbol __getitem__ using list_outputs() is too expensive, when it only cares about the output count in most cases

* Add cython cmake

* GetNumOutputs() and __len__ changes per PR comments

* set commit for tvm
2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 8d960c4bb5 [FRONTEND] Fix mxnet multi outputs (#271)
* fix mxnet multi outputs

* add test case for multi outputs

* fix

* fix

* fix

* fix index

* use json hack

* fix test cases

* fix test cases

* fix test cases

* fix test cases

* fix test cases

* fix test cases
2018-05-29 08:47:00 -07:00
Cyril Lashkevich 8d39aabaa2 Added support for CoreML Permute layers (#262)
https://apple.github.io/coremltools/generated/coremltools.models.neural_network.html?highlight=permute#coremltools.models.neural_network.NeuralNetworkBuilder.add_permute
2018-05-29 08:47:00 -07:00
ziheng ddbbacc3c7 [CMPL] Add Support for Other Data Types (#252)
* [CMPL] Add Support for Other Data Types

* [CMPL] Add test

* [CMPL] Fix
2018-05-29 08:47:00 -07:00
masahi 97aaadeb81 fix onnx conv2d_transpose loading (#245) 2018-05-29 08:47:00 -07:00
ziheng 51a2598267 [FIX] Fix from_mxnet for multiple outputs symbol (#247) 2018-05-29 08:47:00 -07:00
Lianmin Zheng cc7cbbe791 [BUILD] add target_host to compiler.build (#240) 2018-05-29 08:47:00 -07:00
srkreddy1238 2b15684f80 Correction in C++sample code. (#238)
Variable declaration correction.
2018-05-29 08:47:00 -07:00
Leyuan Wang a6182eaad9 unroll condition changed for cuda (#232) 2018-05-29 08:47:00 -07:00
abergeron a0b0546a29 Add conda package (#231)
* Make sure to install the nnvm_compiler library.

* Add conda packge.
2018-05-29 08:47:00 -07:00
Tianqi Chen c6c2e15663 [CMAKE] Update cmake to support OSX/Linux (#228) 2018-05-29 08:47:00 -07:00
ziheng 510105fc99 [FIX] Update rasp benchmark example (#225) 2018-05-29 08:47:00 -07:00
WolffyChen d7cc78a6cf Update deploy.md (#223)
* Update deploy.md

Add an example in c++.

* Update deploy.md

Add an example in c++.

* Update deploy.md

Add an example in c++.

* Update deploy.md

Add an example in c++.

* Update deploy.md

Update dlpack device and type enumerations.
2018-05-29 08:47:00 -07:00
Yuwei Hu 90e4dd6e6c [TOP] fix weight layout in conv2d_transpose (#220)
* update tvm

* [TOP] fix weight layout in conv2d_transpose
2018-05-29 08:47:00 -07:00
WolffyChen 249f42f417 Update deploy.md (#219)
* Update deploy.md

Add an example in c++.

* Update deploy.md

Add an example in c++.

* Update deploy.md

Add an example in c++.

* Update deploy.md

Add an example in c++.
2018-05-29 08:47:00 -07:00
Yuwei Hu 10ca0b1fca [TOP] add conv2d_transpose (#217)
* [TOP] add conv2d_transpose

* update tvm

* fix pylint
2018-05-29 08:47:00 -07:00
Joshua Z. Zhang ad7ffd35bb add softmaxoutput (#207) 2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 8b6f80c262 [Frontend] ONNX frontend v0.2 support (#202)
* update

* fix

* use generated onnx model

* fix tests

* fix lint

* remove log filter

* add vgg

* fix tests

* update tests

* fix download

* fix ci

* fix tutorial url

* clean cache
2018-05-29 08:47:00 -07:00
tqchen f5d158b7fe [TOP] Support ceil_mode 2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 67820febf6 fix version comparison (#200) 2018-05-29 08:47:00 -07:00
Tianqi Chen 920d498d5d More comment change on bench script (#197) 2018-05-29 08:47:00 -07:00
Tianqi Chen 1e49a92dd0 [TVM] Update tvm and benchmark script (#196) 2018-05-29 08:47:00 -07:00
Takeo Imai e545c9a6b7 Change as graph.input can be either ValueInfoProto or string (#186) (#192)
* graph.input can be either ValueInfoProto or string

* pylint
2018-05-29 08:47:00 -07:00
Takeo Imai e7bbb32c4b Change the way to import ONNX model. (#186, #187) (#189)
* Change the way to import ONNX model.

* Decide ONNX version and change the processing

* Check whether onnx has the attribute '__version__'
2018-05-29 08:47:00 -07:00
PENGUINLIONG ae16a366d7 Fixed attribute parameter (#184) 2018-05-29 08:47:00 -07:00
PENGUINLIONG 3833ed2ee9 [FRONTEND] Correct the use of `concatenate` operator (#181)
* Correct the use of `concatenate` operator

* Optimize
2018-05-29 08:47:00 -07:00
ziheng c0cde10d57 [OPT] Improve PreComputePrune When Output Is Pruned (#178) 2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 9a6feca6d2 [FRONTEND] Composed operators (#175)
* fix for composed symbol

* fix

* clean up

* fix exception type
2018-05-29 08:47:00 -07:00
Tianqi Chen 9fb13a6918 [TVM] upgrade to generic schedule (#173) 2018-05-29 08:47:00 -07:00
Tianqi Chen 08e71b73c7 Fix lint, temporary add amaga back (#170) 2018-05-29 08:47:00 -07:00
Tianqi Chen 4dea1502eb [BUILD] Make the core library compatible with msvc 13 (#167) 2018-05-29 08:47:00 -07:00
Eric Junyuan Xie a1404e236f fix symbol output compose (#166) 2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 35a7eac5cc [Frontend] Onnx improvement (#165)
* fix recently released layers

* fix fc layers with partial infer_shape
2018-05-29 08:47:00 -07:00
PENGUINLIONG 51e7851689 Fix Cmake error on Windows (#160) 2018-05-29 08:47:00 -07:00
ziheng 7396297791 [FIX] Fix Typo (#164) 2018-05-29 08:47:00 -07:00
Tianqi Chen 010953b4d1 [RUNTIME] Remove header dependency on private header (#163) 2018-05-29 08:47:00 -07:00
Tianqi Chen 8f4dcd54db [CMAKE] Windows build instruction (#161) 2018-05-29 08:47:00 -07:00
健美猫 90554d08c8 Install nnvm lib and haders. Offer choice to build static or shared lib. (#146) 2018-05-29 08:47:00 -07:00
Leyuan Wang f5bb0e02ae cuda imagenet inference benchmark timing updated (#158) 2018-05-29 08:47:00 -07:00
Tianqi Chen 08023bd949 Update README.md 2018-05-29 08:47:00 -07:00
Tianqi Chen eb8931b4a5 Update README.md 2018-05-29 08:47:00 -07:00
Joshua Z. Zhang aa9528b3f6 [FRONTEND] CoreML (#63)
* add coreml

* fix bool

* fix flatten in fullyconnected

* fix duplicate flatten

* fix syntax

* add tutorial

* fix mxnet flatten, fix tutorials

* fix flatten issue

* fix lint
2018-05-29 08:47:00 -07:00
Leyuan Wang 30157d821f Add cuda imagenet inference benchmark script (#73)
* resnet example merged to imagenet

* resnet example merged to imagenet

* merge with master

* merge with master

* cuda imagenet benchmakr added

* benchmark syntax fixed
2018-05-29 08:47:00 -07:00
Tianqi Chen f72540d7fc fix coreml tools (#72) 2018-05-29 08:47:00 -07:00
Tianqi Chen 9de2e18794 [TEST] include coreml tools in docker (#71)
* [TEST] include coreml tools in docker

* upgrade tvm
2018-05-29 08:47:00 -07:00
ziheng c56a221621 [BENCH] Add Benchmark for Rasp (#68)
* [BENCH] Add Benchmark for Rasp

* [BENCH] Add arg opt-level

* [BENCH] Add model choices

* [BENCH] Improve
2018-05-29 08:47:00 -07:00
Tianqi Chen c53180c722 [GRAPHIR] Print attributes of input (#69) 2018-05-29 08:47:00 -07:00
Tianqi Chen fafa768b13 [DOCS] update docs (#67) 2018-05-29 08:47:00 -07:00
Tianqi Chen 4558400424 [COMPILE] More debug message when compile error (#66) 2018-05-29 08:47:00 -07:00
ziheng f5a8a5c7d7 [TUTORIAL] Deploy the Pretrained Model on Raspberry Pi (#61)
* [TUTORIAL] Deploy the Pretrained Model on Raspberry Pi

* [TUTORIAL] Improve

* [TUTORIAL] Improve

* [TUTORIAL] Improve

* [TUTORIAL] Improve
2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 08ec2de8d5 [DOC] add json spec intro (#62)
* add json spec intro

* fix error

* fix name attrs

* rename and move to top

* address boolean values

* update title
2018-05-29 08:47:00 -07:00
Joshua Z. Zhang f7572dd5f5 [TUTORIAL] Onnx tutorial (#60)
* update

* back to color
2018-05-29 08:47:00 -07:00
ziheng f8db4af6c4 [OP] Conv2d and Depthwise Conv2d for Raspberry Pi (#49)
* [TUTORIAL] ImageNet Inference on Raspberry Pi

* Update tvm
2018-05-29 08:47:00 -07:00
Tianqi Chen 0aafbff4ee Fix docs issue (#59) 2018-05-29 08:47:00 -07:00
Tianqi Chen e5557b04b5 use shorter title (#58) 2018-05-29 08:47:00 -07:00
Tianqi Chen 7fffbeb33d [DOCS] Add install docs, fix Jenkins (#57) 2018-05-29 08:47:00 -07:00
Tianqi Chen d25138e6c9 [PASS] FoldScaleAxis (#55)
* [PASS] FoldScaleAxis

* Move FoldAxis to O3

* Set unroll to 0 when ready
2018-05-29 08:47:00 -07:00
Xingjian Shi 3f599a60be add squeeze (#52)
* add transform

* fix

* update doc

* Update tvm
2018-05-29 08:47:00 -07:00
Tianqi Chen 5541a27502 [TEST] Xavie initialization for benchmarks (#54)
* [TEST] Xavie initialization for benchmarks

* remove additional line
2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 9101181bf2 revert v1 (#53) 2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 092150194b [TUTORIAL] use resnet v2 (#51)
* use resnet v2

* fix
2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 120753d4fa [TUTORIAL] Onnx tutorial (#50)
* add onnx

* fix
2018-05-29 08:47:00 -07:00
Tianqi Chen b65c221282 [TEST] Checkin test docker and scripts (#48) 2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 950aa1a0c2 [Tutorial] mxnet (#47)
* [Tutorial] mxnet

update

add from_gluon

add to __init__

fix tutorial and from_gluon

fix doc lint

merge from_mxnet

fix

fix

fix tutorial

fix

fix header

* fix tutorial

* fix data

* fix
2018-05-29 08:47:00 -07:00
Leyuan Wang de027d94f2 Merge with pull request #44 (#46)
* resnet example merged to imagenet

* merge with master
2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 4f664f5bd4 [Frontend] Onnx (#40)
* init onnx

finish onnx frontend

add onnx tests

fix various

backup

use transformer

[Frontend] graph passed

add test forward

test forward

fix doc and lint

fix test graph tuple

from_onnx now take 2 args, output (sym, params)

fix rename

fix input names

fix multiple

fix lint

fix lint check

* better doc
2018-05-29 08:47:00 -07:00
Tianqi Chen dddd8d1a54 [DOCS] Add intro tutorial (#45) 2018-05-29 08:47:00 -07:00
Tianqi Chen 505f9dd1d4 [TOP] split, reshape, concatenate (#43) 2018-05-29 08:47:00 -07:00
Tianqi Chen f21b5cab7f [DOCS] Add save_param_dict, readme (#42) 2018-05-29 08:47:00 -07:00
Yuwei HU b7b0061100 update nnvm.runtime to tvm.contrib.graph_runtime (#41) 2018-05-29 08:47:00 -07:00
tqchen 007a06dbde [FIX] Make master compile 2018-05-29 08:47:00 -07:00
Tianqi Chen 409014460f [RUNTIME] Remove runtime, rely on tvm only (#38) 2018-05-29 08:47:00 -07:00
Tianqi Chen 3a5b9fcbc9 [DOCS][FRONTEND] Modify from_mxnet to also return params, update docs (#36) 2018-05-29 08:47:00 -07:00
Tianqi Chen 34d74282ec [TUTORIAL] Move mobilenet to tutorial, fix precompute_prune (#35)
* [TUTORIAL] Move mobilenet to tutorial, fix precompute_prune

* Some language improvements
2018-05-29 08:47:00 -07:00
Tianqi Chen 12fa91487d [DOCS] Revamp the docs (#34) 2018-05-29 08:47:00 -07:00
Yuwei HU cba957e09e [TOP][Example] register pool, global_pool; add mobilenet example (#32)
* register pool, global_pool; add mobilenet example

* tests of pool and global_pool

* use new API of runtime module

* small fix
2018-05-29 08:47:00 -07:00
Tianqi Chen 942d8b0ef6 [ATTR] More robust attr parsing (#33) 2018-05-29 08:47:00 -07:00
Tianqi Chen d781a57f42 [RUNTIME] Minimum runtime module (#31) 2018-05-29 08:47:00 -07:00
Tianqi Chen 343c19a5bb [COMPILER] GraphHash based cache system, allow dump and query duplicated functions. (#30) 2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 300ae30aca fix lint (#29) 2018-05-29 08:47:00 -07:00
Tianqi Chen 8aa0100064 [TOP][COMPILER] sum, min, max, transpose, fix dense (#28) 2018-05-29 08:47:00 -07:00
Joshua Z. Zhang 389a00f604 init mxnet converter (#27)
graph

backup

update

finish mxnet converter

fix

fix various

add tests

fix

add multi networks

uses model_zoo

fix tests

minor fix

fix graph

fix
2018-05-29 08:47:00 -07:00
Tianqi Chen 2b3d2e21b7 [PASS] Improve GraphFuse to include five patterns (#26) 2018-05-29 08:47:00 -07:00
Tianqi Chen 2e9b6b9959 [TOP][COMPILER] Add expand_dims, change graph_compare to not compare input optionally (#25) 2018-05-29 08:47:00 -07:00
Tianqi Chen 40bc10f33b [PASS] SimplifyBatchNorm->SimplifyInference, remove dropout (#24) 2018-05-29 08:47:00 -07:00
Tianqi Chen 215693df04 [TOP] Add dense, batchnorm (#22)
* [TOP] Add dense, batchnorm

* update tvm
2018-05-29 08:47:00 -07:00
Tianqi Chen a2ab3d83a7 [RUNTIME][COMPILER] Formal compiler pipeline, runtime wrapper module (#21)
* [RUNTIME][COMPILER] Formal compiler pipeline, runtime wrapper module

* more detailed comments
2018-05-29 08:47:00 -07:00
Tianqi Chen 79ceb9f728 [PASS] PrintGraphIR Join attributes when print ir (#20) 2018-05-29 08:47:00 -07:00
Tianqi Chen c829bd865f [PASS] PrintGraphIR, SimplifyBatchNormInference (#19) 2018-05-29 08:47:00 -07:00