~\;jD"dZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z dd lm Zgd Zd d d ddddZddZej$ej&ej(ej$ej&ej*dZddZiZdZiZdZe j8e j8ej8ej8ej8dZe j<e j<ej<ej<ej<dZdZdZdZ y)z Handles dispatching array operations to the correct backend library, as well as converting arrays to backend formats and then potentially storing them as constants. N) object_arrays)cupy)jax) tensorflow)theano)torch)get_func has_einsum has_tensordotbuild_expressionevaluate_constants has_backendz dask.arrayz theano.tensorzopt_einsum.backends.torchz jax.numpyzautograd.numpyz mars.tensor)daskrr rautogradmarsc tjtj||}| t ||St |||S#t $rd}t |j ||wxYw)zTry and import ``{backend}.{func}``. If library is installed and func is found, return the func; otherwise if default is provided, return default; otherwise raise an error. z{} doesn't seem to provide the function {} - see https://optimized-einsum.readthedocs.io/en/latest/backends.html for details on which functions are required for which contractions.) importlib import_module_aliasesgetgetattrAttributeErrorformat)funcbackenddefaultlib error_msgs eG:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\opt_einsum/backends/dispatch.py _import_funcr!sr >%%hll7G&DE%,_wsD!U'#tW:UU >[ Y--gt<== >s6A A'A-)) tensordotnumpy) transposer#)einsumr#)r"object)r$r&)r%r&cl t||fS#t$rt|||}|t||f<|cYSwxYw)ziReturn ``{backend}.{func}``, e.g. ``numpy.einsum``, or a default func if provided. Cache result. ) _cached_funcsKeyErrorr!)rrrfns r r r <sFT7]++  $ 1') dGm$ s #33c t|S#t$r; td|dt|<n#t$r dt|<YnwxYwt|cYSwxYw)zHCheck if ``{backend}.einsum`` exists, cache result for performance. r%TF) _has_einsumr)r rrs r r r Ls` $7## $ ) Xw '#'K  )#(K  )7##$- A+AAAA AAc t|S#t$r; td|dt|<n#t$r dt|<YnwxYwt|cYSwxYw)zKCheck if ``{backend}.tensordot`` exists, cache result for performance. r"TF)_has_tensordotr)r rr-s r r r ^s` 'g&& ' , [' *&*N7 # ,&+N7 # ,g&&'r.)rrrr rc t|||S)zxBuild an expression, based on ``expr`` and initial arrays ``arrays``, that evaluates using backend ``backend``. )CONVERT_BACKENDSrarraysexprs r r r s G $VT 22c t|||S)zConvert constant arrays to the correct backend, and perform as much of the contraction of ``expr`` with these as possible. )EVAL_CONSTS_BACKENDSr3s r rrs  ( 66r6c.|jtvS)z$Checks if the backend is known. )lowerr2r-s r rrs ==?. ..r6)N)r#N)!__doc__rr#rr_cupyr_jaxr _tensorflowr_theanor _torch__all__rr!r"r$r% object_einsumr(r r,r r0r r r2rr8rr6r rEs  ' n  (     >&"OO!OO"__"__'55    $ '$..&&  " "  $ $  00((  $ $  & &  " " 37/r6