• Z
    fix input shape of op tests (#21682) · 4c987a60
    zhupengyang 提交于
    * fix input shape of op tests for elementwise_sub,
    gather, pad2d, transpose, softmax, scale,
    elementwise_max, hierarchical_sigmoid, reshape2,
    sign, squeeze, reduce_sum, sum, squeeze2,
    unsqueeze, unsqueeze2, cast, reverse
    
    test=develop
    
    * fix cast, elementwise_mul, gather, scale, sign,
    softmax, transpose
    
    test=develop
    4c987a60
test_unsqueeze_op.py 2.2 KB