projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: df6f54a) | patch
[Relay] Add new IR pass CombineParallelDense (#3862)
authorJon Soifer <soiferj@gmail.com>
Tue, 24 Sep 2019 08:12:11 +0000 (01:12 -0700)
committerJared Roesch <roeschinc@gmail.com>
Tue, 24 Sep 2019 08:12:11 +0000 (01:12 -0700)
commited9fdfb0926f56990df4bb9768774b9d32a43b39
tree45d6b675ba92285613b8d8c7491c1ee8de45cbc0tree | snapshot
parentdf6f54ac9ab86fb7971b12e5aad451cf2949dda3commit | diff
[Relay] Add new IR pass CombineParallelDense (#3862)

* Refactor to create abstract ParallelOpCombiner

* First draft of CombineParallelDense

* Begin to work on tests

* Test

* Refactor to move out more common code

* Clean up

* Fix

* Remove statics

* fix wording

* Start to add combine_parallel_op_batch

* Resolve PR comments

* Resolve PR comments

* dummy change to retrigger CI

* Change special case from bias_add to add

* Revert special case change

* Ignore units check

* dummy change to retrigger CI

* dummy change to re-trigger CI

* Improve docs

* Update docs

* Update docs
12 files changed:
docs/api/python/relay/transform.rst diff | blob | history
include/tvm/relay/transform.h diff | blob | history
python/tvm/relay/transform.py diff | blob | history
src/relay/backend/build_module.cc diff | blob | history
src/relay/pass/combine_parallel_conv2d.cc diff | blob | history
src/relay/pass/combine_parallel_dense.cc [new file with mode: 0644] blob
src/relay/pass/combine_parallel_op.cc [new file with mode: 0644] blob
src/relay/pass/combine_parallel_op.h [new file with mode: 0644] blob
src/relay/pass/combine_parallel_op_batch.cc [new file with mode: 0644] blob
src/relay/pass/combine_parallel_op_batch.h [new file with mode: 0644] blob
src/relay/pass/pattern_util.h diff | blob | history
tests/python/relay/test_pass_combine_parallel_dense.py [new file with mode: 0644] blob
Domain: Machine Learning / ML Framework;