\;j$ddlZddlZddlmZddlZddlZddlmZddl m Z m Z m Z m Z mZmZddlmZeeej&dZe ZeZe j.Ze Ze Ze ZdZd Zd efd ed ed edededede deefdZ!ee_"ee_#y)N)Set)core)AnalysisConfig PaddleDTypePaddleInferPredictorPaddleInferTensor PaddlePlaceconvert_to_mixed_precision_bind) get_loggerz&%(asctime)s-%(levelname)s: %(message)s)fmtct|tjs1t|tr3t |dkDr%t|dt r|j |ytd)zA Support input type check based on tensor.copy_from_cpu. rzHIn copy_from_cpu, we only support numpy ndarray and list[str] data type.N) isinstancenpndarraylistlenstr_copy_from_cpu_bind TypeErrorselfdatas aG:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddle/inference/wrapper.pytensor_copy_from_cpur-sQ$ #43t9q=ZQ5M   & V  c:t|tjr|j|yt|tj r|j |yt|tjjjr tdtd)zG Support input type check based on tensor.share_external_data. zThe interface 'share_external_data' can only be used in dynamic graph mode. Maybe you called 'paddle.enable_static()' and you are in static graph mode now. Please use 'copy_from_cpu' instead.z=In share_external_data, we only support Tensor and LoDTensor.N) rr LoDTensor_share_external_data_bindpaddleTensor'_share_external_data_paddle_tensor_bindbase frameworkVariablerrs rtensor_share_external_datar%;sz$' &&t, D&-- ( 44T: D&++//88 9 2   K  rT model_file params_filemixed_model_filemixed_params_filemixed_precisionbackend keep_io_types black_listc |tjur)tjstj dt jj|} t|dk7rt jj|n| } t jj| st j| |jdt} t||||||||| y)a Convert a fp32 model to mixed precision model. Args: model_file: fp32 model file, e.g. inference.pdmodel. params_file: fp32 params file, e.g. inference.pdiparams. mixed_model_file: The storage path of the converted mixed-precision model. mixed_params_file: The storage path of the converted mixed-precision params. mixed_precision: The precision, e.g. PrecisionType.Half. backend: The backend, e.g. PlaceType.GPU. keep_io_types: Whether the model input and output dtype remains unchanged. black_list: Operators that do not convert precision. kwargs: Supported keys including 'white_list'. - white_list: Operators that do convert precision. zOYou shoule use PaddlePaddle compiled with GPU when backend set to PlaceType.GPUr white_listN) PlaceTypeGPUris_compiled_with_cuda_loggererrorospathdirnamerexistsmakedirsgetsetr ) r&r'r(r)r*r+r,r-kwargsmixed_model_dirnamemixed_params_dirnamer/s rconvert_to_mixed_precisionr?Os4)--(B(B(D ] ''//*:;  !Q & )*  77>>. / ()L#%0J# r)$loggingr5typingrnumpyrr paddle.baserpaddle.base.corerrrrr r paddle.base.log_helperr __name__INFOr3DataTyper0 Precision PrecisionTypeConfigr Predictorrr%r;rboolr? copy_from_cpushare_external_datarrrQs  .  gll H    ((      655555 5 # 5  55C5p,7r