Bvh lddlmZddlZddlZddlZddlZddlZddlmZm Z m Z dZ e dk(re yy)) annotationsN)fix_output_shapesmake_dim_param_fixedmake_input_shape_fixedc4tjtjj t dt jd}|jdtdd|jdtdd |jd tdd |jd d dd|jdtjd|jdtjd|j}|jr |jss|js |jr[|jr|j rC|j dks4|jrX|j"rt%d|j"Dr0t'd|j)t+j,dt/j0t|j2j5d}|jr,t7|j8|j|j n+t;|j8|j|j"t=|t/j>|t|j@j5y)N:z Assign a fixed value to a dim_param or input shape Provide either dim_param and dim_value or input_name and input_shape.) descriptionz --dim_paramFz8Symbolic parameter name. Provide dim_value if specified.)typerequiredhelpz --dim_valuez:Value to replace dim_param with in the model. Must be > 0.z --input_namezGModel input name to replace shape of. Provide input_shape if specified.z --input_shapec\|jdDcgc] }t|c}Scc}w)N,)splitint)xis T/RAG/venv/lib/python3.12/site-packages/onnxruntime/tools/make_dynamic_shape_fixed.pyz1make_dynamic_shape_fixed_helper..&s" 5 1A 55s)zShape to use for input_shape. Provide comma separated list for the shape. All values must be > 0. e.g. --input_shape 1,3,256,256 input_modelz%Provide path to ONNX model to update.)r r output_modelz,Provide path to write updated ONNX model to.rc3&K|] }|dk yw)rN).0values r z2make_dynamic_shape_fixed_helper..5srAs:#  ]]57p z#%r=