Ti%SrSSKJr /SQrSSKrSSKJrJr SSKJr SSK r SSK J r \(aSSK r SqS\S'SqS \S 'SS jrSS jr"S S\ R&5rg)aCompatible adapters implementing the TensorProtocol interface for various framework tensor types. This module provides public classes that implement the :class:`onnx_ir.TensorProtocol` interface for various tensor types from popular deep learning frameworks. You can use these classes to create tensors and use them in the IR graph like any other tensor. Example:: import torch import onnx_ir as ir # Create a PyTorch tensor torch_tensor = torch.tensor([1, 2, 3]) # Wrap the PyTorch tensor in a TorchTensor object ir_tensor = ir.tensor_adapters.TorchTensor(torch_tensor) # Use the IR tensor in the graph attr = ir.AttrTensor("x", ir_tensor) print(attr) ) annotations)from_torch_dtypeto_torch_dtype TorchTensorN) TYPE_CHECKINGAny)_corez%dict[torch.dtype, ir.DataType] | None_TORCH_DTYPE_TO_ONNXz%dict[ir.DataType, torch.dtype] | None_ONNX_DTYPE_TO_TORCHc [GcySSKn0UR[RR _UR [RR_UR[RR_UR[RR_UR[RR_UR[RR_UR [RR"_UR$[RR&_UR([RR*_UR,[RR._UR0[RR2_UR4[RR6_UR8[RR:_UR<[RR>_UR@[RRB_URD[RRF_URH[RRJ_URL[RRNURP[RRR0Eq[UUS5(a+[RRV[URX'[UUS5(a+[RRZ[UR\'[UUS5(a+[RR^[UR`'U[;a,[cSUS[e[Rg5535e[U$)z/Convert a PyTorch dtype to an ONNX IR DataType.Nrfloat8_e8m0fnuint2uint2zUnsupported PyTorch dtype 'z/'. Please use a supported dtype from the list: )4r torchbfloat16irDataTypeBFLOAT16boolBOOL complex128 COMPLEX128 complex64 COMPLEX64float16FLOAT16float32FLOATfloat64DOUBLE float8_e4m3fn FLOAT8E4M3FNfloat8_e4m3fnuzFLOAT8E4M3FNUZ float8_e5m2 FLOAT8E5M2float8_e5m2fnuzFLOAT8E5M2FNUZint16INT16int32INT32int64INT64int8INT8uint8UINT8uint16UINT16uint32UINT32uint64UINT64hasattr FLOAT8E8M0r INT2rUINT2r TypeErrorlistkeysdtypers V/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx_ir/tensor_adapters.pyrr5s# NNBKK00 JJ ((    bkk44  OOR[[22  MM2;;..  MM2;;,,   MM2;;--    !9!9   ! !2;;#=#=    r{{55   ! !2;;#=#=  KK**  KK**  KK**  JJ (( KK**! " LL"++,,# $ LL"++,, LL"++,,' * 5* + +9;9O9O !5!5 6 5& ! !/1{{/?/?  , 5' " "02 0A0A  - (()%1;(--/01 3   &&c [GcySSKn0[RRUR _[RR UR_[RRUR_[RRUR_[RRUR_[RRUR_[RR UR"_[RR$UR&_[RR(UR*_[RR,UR._[RR0UR2_[RR4UR6_[RR8UR:_[RR<UR>_[RR@URB_[RRDURF_[RRHURJ_[RRLURN[RRPURR0Eq[UUS5(a+URV[[RRX'[UUS5(a+URZ[[RR\'[UUS5(a+UR^[[RR`'U[;aUU[RRX:Xa [cS5e[eSUS[g[Ri5535e[U$) z/Convert an ONNX IR DataType to a PyTorch dtype.Nrr rrz}The requested DataType 'FLOAT8E8M0' is only supported in PyTorch 2.7+. Please upgrade your PyTorch version to use this dtype.z(Unsupported conversion from ONNX dtype 'z8' to torch. Please use a supported dtype from the list: )5r rrrrrrrrrrrrrrrr rr"r!r$r#r&r%r(r'r*r)r,r+r.r-r0r/r2r1r4r3r6r5r8r7r9r r:rr;rr< ValueErrorr=r>r?r@s rBrras# KK %.. KK  ejj  KK " "E$4$4  KK ! !5??  KK    KK  u}}   KK     KK $ $e&9&9  KK & &(=(=  KK " "E$5$5  KK & &(=(=  KK  u{{  KK  u{{  KK  u{{  KK  ejj KK  u{{! " KK   # $ KK   KK   ' , 5* + +;@;O;O !7!7 8 5& ! !5:ZZ !1!1 2 5' " "6;kk !2!2 3 (( BKK** *I 6ug>;(--/01 3   &&rCcn^\rSrSrS S U4SjjjrS SjrS S SjjrSrS SjrSSjr Sr U=r $)rcJ>[TU]U[UR5X#S9 g)N)rAname doc_string)super__init__rrA)selftensorrIrJ __class__s rBrLTorchTensor.__init__s(  *6<<8t  rCcDSSKnU UR[RR:XaYUR R UR5RSS9R URR55$UR[RR[RR[RR[RR[RR1;aYUR R UR5RSS9R URR55$UR[RR[RR 1;aYUR R UR5RSS9R URR55$UR RSS9$)NrT)force)rrArrrrawviewr3numpyr"r$r&r(r:r1r;r<)rMrs rBrUTorchTensor.numpysc ::-- -88==.4444@EEdjjFVFVFXY Y :: KK $ $ KK & & KK " " KK & & KK " "   88==-33$3?DDTZZEUEUEWX X ::"++**BKK,=,=> >88==-33$3?DDTZZEUEUEWX Xxx~~D~))rCchAUcUR5$UR5RU5$N)rU __array__)rMrAcopys rBrYTorchTensor.__array__s-  =::< zz|%%e,,rCc$SSKnURRR5 URR 5R 5R5nSSS5 [WURRR5(a[SURS35eU[RUR5-UR5-R!UR#554$!,(df  N=f)z4Get a ctypes byte array pointing to the tensor data.rNz.Cannot take content out from the FakeTensor ('z'). Please replace the tensor with a tensor backed by real data using ONNXProgram.apply_weights() or save the model without initializers by setting include_initializers=False.)torch._subclasses.fake_tensor _subclasses fake_tensorunset_fake_temporarilyrSdetachcpu contiguous isinstance FakeTensorr=rIctypesc_ubyte element_sizenumel from_addressdata_ptr)rMrrNs rB _get_cbytesTorchTensor._get_cbytess,    * * A A CXX__&**,779FD fe//;;FF G G@ L`` )<)<)>>O]] OO    D Cs 7D Dc<UR5up[U5$rX)rlbytes)rM_datas rBtobytesTorchTensor.tobytess""$T{rCcHUR5up#URU5$rX)rlwrite)rMfilerprqs rBtofileTorchTensor.tofiles!""$zz$rC)NN)rNz torch.TensorrI str | NonerJrz)return npt.NDArray)rArrZz bool | Noner{r|)r{ro)r{None) __name__ __module__ __qualname____firstlineno__rLrUrYrlrrrw__static_attributes__ __classcell__)rOs@rBrrsFVZ " *4 IS  *&-  (  rCr)rA torch.dtyper{ ir.DataType)rArr{r)__doc__ __future__r__all__rftypingrr numpy.typingnptonnx_irrr rr __annotations__r rrTensorrryrCrBrsd4#  %?C;B>B;B)'X/'d? %,,? rC