test_pass_alter_op_layout.py
17.7 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