projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 552d4aa) | patch
[Relay] InferCorrectLayout for strided_slice & min_num_branches option in CombinePara...
authorWuwei Lin <vincentl13x@gmail.com>
Tue, 9 Apr 2019 05:20:56 +0000 (13:20 +0800)
committerYizhi Liu <liuyizhi@apache.org>
Tue, 9 Apr 2019 05:20:56 +0000 (22:20 -0700)
commit5d70b008668ba244dad598d1ce3e8a79e2d6cede
tree1f6f91b523058bca5541368e2478143375b4d976tree | snapshot
parent552d4aa3587aa3a0443d050ceb324386489ff793commit | diff
[Relay] InferCorrectLayout for strided_slice & min_num_branches option in CombineParallelConv2D (#2961)

* [Relay] InferCorrectLayout for strided_slice

* Add min_num_branches option to CombineParallelConv2D

* Return undef if original layout contains splitted axes
python/tvm/relay/ir_pass.py diff | blob | history
src/relay/op/tensor/transform.cc diff | blob | history
src/relay/pass/combine_parallel_conv2d.cc diff | blob | history
tests/python/relay/test_pass_alter_op_layout.py diff | blob | history
tests/python/relay/test_pass_combine_parallel_conv2d.py diff | blob | history
Domain: Machine Learning / ML Framework;