- Aug 17, 2017
-
-
Tianqi Chen authored
-
Tianqi Chen authored
-
Tianqi Chen authored
-
William Moses authored
-
William Moses authored
-
- Aug 16, 2017
-
-
Tianqi Chen authored
-
ziheng authored
* [NNPack] Support for threadpool * fix lint * fix lint * Use static class function
-
wetliu authored
* rename the nchw and pass the unit test; going to do it for nhwc depthwise * bug with fusion * nchw works fine; nhwc float32 problem remains * still cannot bind them together * fusion works * syntax fix * all bugs fixed; test cases pass * minor fix on nn.h
-
- Aug 15, 2017
-
-
ziheng authored
* [Contrib] CuDNN v7 Support * Add test
-
Tianqi Chen authored
-
Tianqi Chen authored
-
Tianqi Chen authored
-
Leyuan Wang authored
* __init__ updated * pull request updated * build_module added * typo fixed * another typo fixed * conv2d gpu scheduler for two layouts moved to tvm * changes made according to CR * conv2d_nchw formating updated, conv2d_hwcn tests updated * lint error fixed * element wise operator schedule fusing fixed for conv2d * conv2d_nchw topi test added, all resnet workloads now pass * conv compute lint error fixed * fixed python 3 compatibility problem * conv2d tensor input support added, test typo fixed, ir_pass.Simplify changed to util.get_const_int * fixed channel numer < 4 error, also made sure other splitting factor woudn't be 0
-
ziheng authored
* [TOPI] Add ops compute Remove 'compute' and add assert for safety Add document fix lint fix softmax * fix batch norm
-
Tianqi Chen authored
-
Yuwei HU authored
-
Tianqi Chen authored
* [TOPI] Move ewise.h -> elemwise.h * fix test
-
Xingjian Shi authored
[TOPI] Add broadcast and reduce operators
-
Tianqi Chen authored
-
- Aug 14, 2017
-
-
Nicolas Vasilache authored
-
Yuwei HU authored
* add dilation operators * fix pylint * dilate testcases success * n-D tensor dilation * support arbitrary dimension
-
Tianqi Chen authored
* [DOC] Include TOPI in doxygen * update
-
Leyuan Wang authored
* __init__ updated * pull request updated * build_module added * typo fixed * another typo fixed * conv2d gpu scheduler for two layouts moved to tvm * changes made according to CR * conv2d_nchw formating updated, conv2d_hwcn tests updated * lint error fixed * element wise operator schedule fusing fixed for conv2d * conv2d_nchw topi test added, all resnet workloads now pass * conv compute lint error fixed * fixed python 3 compatibility problem * conv2d tensor input support added, test typo fixed, ir_pass.Simplify changed to util.get_const_int
-
Tianqi Chen authored
* [TOPI] Move topi.nn.util to topi.util * update the path
-
- Aug 13, 2017
-
-
Nicolas Vasilache authored
* [WIP] C++ topi contributions Summary: This diff implements C++ topi contributions for: - relu with parametrix threshold - pad with generic padBefore / padAfter specification - matmult with transposes - conv2d_nchw, conv2d_hwcn with runtime constant padding and strides - depthwise_conv2d_nchw with runtime constant padding and strides - group_conv2d_ngchw with runtime constant padding and strides - broadcast_to a broadcastable shape - broadcast_bop where bop is an usual binary op (+ - * / %) Convolution padding is implemented using the pad operation. To avoid extra memory consumption, it is generally recommended to inline the padding with the autoinliner. Unfortunately in its current form the elemwise checks are too restrictive to allow inlining. So this diff also proposes an extension to LHS injective (i.e. no reduction axis in the current IR design) Test Plan: Tested in C++ testsuite in a separate repository, I am looking for suggestions to quickly spin up some tests for tvm. Reviewers: tqchen Subscribers: Tasks: Tags: Blame Revision: * Review + Lint + GSG C++
-
Tianqi Chen authored
-
Tianqi Chen authored
-
- Aug 12, 2017
-
-
Tianqi Chen authored
-
- Aug 11, 2017
-
-
Yuwei HU authored
-
Tianqi Chen authored
* [PASS][FIX] Fix LiftAttrScope with if * [PASS] Fix on proc sync * fix
-
- Aug 10, 2017
-
-
Tianqi Chen authored
-
Yizhi Liu authored
-
Tianqi Chen authored
* [TEST] Upgrade gpu docker to cudnn7 * fx
-
- Aug 09, 2017
-
-
Tianqi Chen authored
-
ziheng authored
-
- Aug 08, 2017
-
-
Tianqi Chen authored
-
Yizhi Liu authored
* [tvm4j] RPC Server * [tvm4j] fix recursively function calling; connect to proxy server; osx rename .so to .dylib * [tvm4j] test case for proxy connection; thread pool for serving
-
Tianqi Chen authored
* [RUNTIME][PASS] Allow declare vector type array * fix bcast * [BUFFER] Enable vload/store function in buffer * ok
-
- Aug 07, 2017
-
-
ziheng authored
* [NNPACK] Add nnpack.convolution * Add instrinsic * Fix lint
-
- Aug 06, 2017
-
-
ziheng authored
* [EXECUTOR] Split graph_executor to header file and (runtime) source file * Fix
-