combine_parallel_conv2d.cc
13.5 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