9jUddlZddlmZmZddlZddlmZdaedzed<ejde dedfdZ d d Z y) N) GeneratorSequence)ContentStoreReaderLOAD_TENSOR_READERlocreturn)NNNc#dKt tdt|da dday#dawxYww)Nz!LOAD_TENSOR_READER is already setF)cache)rAssertionErrorr)rs X/media/conek/DATA/Code/OCR/venv/lib/python3.12/site-packages/torch/_prims/debug_prims.pyload_tensor_readerr s=%@AA,Cu=" !Ts0)0-0c 2tjjddtjjdddtdt t dt t dtjdtjd tjf d }y) Nzdebugprims::load_tensorzR(str name, int[] size, int[] stride, *, ScalarType dtype, Device device) -> Tensor BackendSelectnamesizestridedtypedevicerctddlm}|||||Sddlm}tj ||}t |j|k7rt|jd|t |j|k7rt|jd||j|k7rt|jd||j|k7r |||}|S)Nr) rand_strided) clone_input)rz != )r) rtorch._dynamo.testingrtorch._dynamo.utilsr read_tensorlistrr rrr)rrrrrrrrs r load_tensor_factoryz1register_debug_prims..load_tensor_factory#s  % :feV< < 7#..tF.CAAFFH~%$zdV%<==AHHJ6)$ |4x%@AAxx6!$zfX%>??ww%/H) torchlibrarydefineimplstrrintrrTensor)rs r register_debug_primsr&s MM!\  ]]1?Csm {{     Dr)rN) contextlibcollections.abcrrrtorch.utils._content_storerr__annotations__contextmanagerr#r r&rr r-sW/ 915&-4 "C"I.>$?""""r