tests/python/relay/test_pass_combine_parallel_conv2d.py · 770ac84e74a5d0cb174c1a5402f0752a5a8fbecb · wenyuanbo / tic

[Relay] InferCorrectLayout for strided_slice & min_num_branches option in… · 5d70b008

[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

committed Apr 08, 2019

5d70b008

test_pass_combine_parallel_conv2d.py 7.96 KB