o ü* iQã@sàddlmZddlmZddlmZmZmZmZm Z m Z m Z m Z ddl mZddlmZmZmZddlmZddlmZdd lmZmZmZmZmZmZdd lmZm Z m!Z!dd l"m#Z#dd l$m%Z%m&Z&m'Z'm(Z(m)Z)gd ¢Z*dS)é)Ú LayerHelper)Úfleeté)ÚaspÚautogradÚautotuneÚccÚlayersÚmultiprocessingÚnnÚxpu)Úauto_checkpoint)Ú get_rng_stateÚregister_rng_state_as_indexÚ set_rng_state)Ú inference)Ú identity_loss)Úgraph_khop_samplerÚ graph_reindexÚgraph_sample_neighborsÚgraph_send_recvÚsoftmax_mask_fuseÚ softmax_mask_fuse_upper_triangle)ÚDistributedFusedLambÚ LookAheadÚ ModelAverage)Úfuse_resnet_unit_pass)Ú _npu_identityÚ segment_maxÚ segment_meanÚ segment_minÚ segment_sum)rrrrrrrrr!rrr rrrN)+Zbase.layer_helperrÚ distributedrÚrrrrr r r r Ú checkpointr Z frameworkrrrZjitrZnn.lossrÚ operatorsrrrrrrZ optimizerrrrZpassesrZtensorrrrr r!Ú__all__©r'r'úd/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/paddle/incubate/__init__.pyÚs  (