U T?h:@s`ddlmZddlmZmZddlmZddlmZddl m Z edddZ edd d Z d S) ) NumpyHelper) ModelProto TensorProto)set_external_data) OnnxModel)OrtValue)modelcCs|g}t|}|D]^}|jD]R}|j}|drt|}t|}| ||ft |dd||_| dqqt |S)a Extract external data from model and return the external data as a list of tuples (name, value). Note this function does not handle external data that is not loaded into the model as raw data. Args: model (ModelProto): the model proto to extract external data from. Returns: (external_names, external_values): a tuple of two lists of external data names and values. raw_datazfoo.bin)location) rgraphs initializernameHasFieldrZto_arrayrZortvalue_from_numpyappendrZ ClearFieldzip)rZ external_data onnx_modelgraphr r Z numpy_tensorZ ort_valuerU/var/www/html/venv/lib/python3.8/site-packages/onnxruntime/transformers/onnx_utils.pyextract_raw_data_from_model s       rcCsDt|}|D].}|jD]"}|dr|jtjkrdSqqdS)z Check if the model has external data. Args: model (ModelProto): the model proto to check for external data. Returns: bool: True if the model has external data, False otherwise. data_locationTF)rr r rrrZEXTERNAL)rrrr rrrhas_external_data)s    rN) Z fusion_utilsrZonnxrrZonnx.external_data_helperrrrZ onnxruntimerrrrrrrs