Ph ddlmZddlZddlmZmZddlZddlmZmZeje Z ejjddf d!dZd"dZedd d ej j"f Zed d dej j"f ZedddZedddZedddZededZedddZedd#dd Zy)$) annotationsN)OptionalSequence)_primsTensorcDfd}tj|||||S)Nc:tj|i|SN)r TensorMeta)argskwargs impl_atens iC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/_inductor/inductor_prims.pymetazmake_prim..metas  D!;F!;<<)schema return_typerrdoctags)r _make_prim)rrrrrrs ` r make_primr s/=        rc|j|k(r|S|jj|j|}|j ||Sr )strideclone as_stridedshapecopy_) input_tensorr new_tensors reager_force_strider! sT&##%00J\" rz&inductor_seed(Device device) -> Tensorc2tjdg|SNldevicetorchrandintr$s rr).s5==Bv>rz=create a fresh seed (one per call) for use with inductor_rand)rrz2inductor_seeds(int count, Device device) -> Tensorc4tjd|g|Sr#r&)countr%s rr)r)4s%-- E76Jrz/Horizontal fusion of many inductor_seed() callsz7inductor_lookup_seed(Tensor seeds, int index) -> Tensorc ||Sr )seedsindexs rr)r);surz9Extract a single seed from the result of inductor_seeds())rz?inductor_random(SymInt[] size, Tensor seed, str mode) -> TensorcFtt|||jSNr$)getattrr'r%)sizeseedmodes rr)r)@s1WUD1$t{{KrzGtorch.rand()/torch.randn() using backend-specific RNG that can be fusedzOinductor_randint(SymInt low, SymInt high, SymInt[] size, Tensor seed) -> TensorcHtj||||jSr1)r'r(r%)lowhighr3r4s rr)r)Es%--T4 "Trz TensorznForce the stride order for input tensor. No-op if the input tensor already has the stride. Do a copy otherwiseziinductor_masked_scatter_with_index(Tensor input, Tensor mask, Tensor source_idx, Tensor source) -> Tensorc0tj|||Sr )r'masked_scatter)rmaskr/sources rr)r)Ose.B.BdF/rz'masked_scatter with precomputed indiceszj_unsafe_index_put_(Tensor(a!) self, Tensor?[] indices, Tensor values, bool accumulate=False) -> Tensor(a!)cZtjjj||||Sr )r'opsaten index_put_)selfindicesvalues accumulates rr)r)Vs EIINN4M4M gvz5rz0Unsafe index_put_ (doesn't issue device asserts))rstrrrErzOptional[Sequence[torch.Tag]])rrreturnr)F) __future__rloggingtypingrrr'rr getLogger__name__log RETURN_TYPENEWrr!Tagnondeterministic_seededr4r. lookup_seedrandomr(force_stride_ordermasked_scatter_with_index_unsafe_index_put_r-rrrVs6"% g! ""&&*.    ( (,>G )) + + -  8J9 )) + + -   =%C  EKQ  UTF  Jx &o 2 p ; r