• Ted Themistokleous's avatar
    Change Unsqueeze to squeeze to parse_if for trailing dimensions · b5d1db2e
    Ted Themistokleous authored
    Default to smaller dimension with trailing 1 case instead of unsqueezing to
    the larger dimension.
    
    More analysis on other networks concludes that when putting in two operands to
    and IF block, the output should take the smaller of the shapes instead of the larger
    
    Modified tests in onnx_test.cpp to parse to the correct output as well.
    b5d1db2e
parse_if.cpp 7.86 KB