• Dean Wyatte's avatar
    Export TensorFlow models to ONNX with dynamic input shapes (#19255) · a26d71d6
    Dean Wyatte authored
    * validate onnx models with a different input geometry than saved with
    
    * only test working features for now
    
    * simpler test skipping
    
    * rm TODO
    
    * expose batch_size/seq_length on vit
    
    * skip certain name, feature, framework parameterizations known to fail validation
    
    * Trigger CI
    
    * Trigger CI
    a26d71d6
test_onnx_v2.py 18 KB