fPi xddlmZddlZddlZddlZddlZddlZddlmZm Z m Z dZ e dkr e dSdS)) annotationsN)fix_output_shapesmake_dim_param_fixedmake_input_shape_fixedctjtjt dt jd}|dtdd|dtdd |d tdd |d d dd|dtj d|dtj d| }|jr|jsS|js|jrE|jr|jr7|jdks,|jr\|jrt%d|jDr7t'd|t+jdt/jt|jd}|jr!t7|j|j|jn t;|j|j|jt=|t/j|t|j dS)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@d|dDS)Nc,g|]}t|S)int).0is ~/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/onnxruntime/tools/make_dynamic_shape_fixed.py zEmake_dynamic_shape_fixed_helper....&s5551A555,)split)xs rz1make_dynamic_shape_fixed_helper..&s55 555rzShape 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|] }|dkV dS)rNr)rvalues r z2make_dynamic_shape_fixed_helper..5s&rCs#"""""  ]]]]]]]]]]575757p z##%%%%%r