test_pass_combine_parallel_conv2d.py
7.96 KB
-
[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
Wuwei Lin committed