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