ó àÆ÷Xc@`s(ddlmZmZmZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZddlmZdefd „ƒYZd efd „ƒYZd efd „ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdS(i(tabsolute_importtprint_functiontdivisionN(tDefaultOrderedDict(t OrderedSet(tStringIOt integer_types(topt(tconfigtDBcB`sheZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z e j d „Z RS( cC`s<t|dƒs5tjd|_tjdcd7}||jkrhtd||ƒ‚n|j|j|ƒq>WdS(Nis+The tag of the object collides with a name.(RtlenRtcopytpopRRR(R RR!R ttag((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyRIs  cG`s„|j|}t|ƒdks%t‚|jƒjƒ}xF|D]>}||jkrhtd||ƒ‚n|j|j|ƒq>WdS(Nis+The tag of the object collides with a name.(RR$RR%R&RRtremove(R RR!R R'((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyt remove_tagsSs  c C`slt|tƒs!td|ƒ‚ntƒ}x%|jD]}|j|j|ƒq4Wx%|jD]}|j|j|ƒq\Wx%|j D]}|j |j|ƒq„Wtƒ}tƒ}x—|D]}t|t ƒr»|}|j rút j |ƒ}g|_ n|jj|j|ƒ}|j|ƒ} |j| _|j|ƒ|j| ƒq»q»W|j |ƒ|j|ƒ|S(NsExpected a Query.(RtQueryRRtincludetupdateRtrequiretintersection_updatetexcludetdifference_updateR textra_optimizationsR%tsubquerytgetRtqueryR( R tqt variablesR'R(RR t def_sub_querytsqt replacement((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyt __query__]s2          c O`sMt|ƒdkrct|dtƒrct|ƒdks=|rRtd||ƒ‚n|j|dƒSg|D]}|jdƒrj|d^qj}g|D]}|jdƒr–|d^q–}g|D]}|jdƒrÂ|d^qÂ}t|ƒt|ƒt|ƒt|ƒkr%td|ƒ‚n|jtd|d |d |d |ƒƒS( NiisNIf the first argument to query is a Query, there should be no other arguments.t+t&t-sIAll tags must start with one of the following characters: '+', '&' or '-'R+R-R/R2(R$RR*RR:t startswithR(R R!tkwtagsR'R+R-R/((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyR4zs%,,,, cC`sa|j|}|s&td|ƒ‚n%t|ƒdkrKtd|ƒ‚nx|D]}|SWdS(NsNothing registered for '%s'is-More than one match for %s (please use query)(RtKeyErrorR$R(R RR6tvariable((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyt __getitem__Œs   cC`s ||jkS(N(R(R R((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyt __contains__–scC`sVtd|jjt|ƒfd|ƒtd|jd|ƒtd|jd|ƒdS(Ns %s (id %i)tfiles namess db(tprintRRtidRR(R tstream((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyt print_summary™s&(Rt __module__R RR#RR)R:R4RBRCtsyststdoutRH(((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyR s   0   R*cB`sbeZdZd d d edƒd d„Zd„Zd„Zd„Zd„Z d„Z d„Z RS( s© Parameters ---------- position_cutoff : float Used by SequenceDB to keep only optimizer that are positioned before the cut_off point. tinfcC`sÇt|ƒ|_|ptƒ|_|p-tƒ|_|p<i|_||_|dkr`g}n||_t|jt t fƒr–t|jƒ|_nt|jt t fƒrÃt|jƒ|_ndS(N( RR+R-R/R2tposition_cutoffRR1Rtlistttuple(R R+R-R/R2RMR1((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyRªs    cC`s,d|j|j|j|j|j|jfS(NsKQuery{inc=%s,ex=%s,require=%s,subquery=%s,position_cutoff=%f,extra_opts=%s}(R+R/R-R2RMR1(R ((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyt__str__ºscC`s/|jj|ƒt|dƒs+g|_ndS(NR1(t__dict__R,R R1(R tstate((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyt __setstate__ÀscG`s4t|jj|ƒ|j|j|j|j|jƒS(N(R*R+tunionR-R/R2RMR1(R R!((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyt includingÆs cG`s4t|j|j|jj|ƒ|j|j|jƒS(N(R*R+R-R/RTR2RMR1(R R!((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyt excludingÏs  cG`s4t|j|jj|ƒ|j|j|j|jƒS(N(R*R+R-RTR/R2RMR1(R R!((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyt requiringØs  c G`s5t|j|j|j|j|j|jt|ƒƒS(N(R*R+R-R/R2RMR1RN(R t optimizations((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyR#às  N( RRIt__doc__RtfloatRRPRSRURVRWR#(((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyR*Ÿs      t EquilibriumDBcB`s/eZdZeed„Zd„Zd„ZRS(s« A set of potential optimizations which should be applied in an arbitrary order until equilibrium is reached. Canonicalize, Stabilize, and Specialize are all equilibrium optimizations. Parameters ---------- ignore_newtrees If False, we will apply local opt on new node introduced during local optimization application. This could result in less fgraph iterations, but this doesn't mean it will be faster globally. tracks_on_change_inputs If True, we will re-apply local opt on nodes whose inputs changed during local optimization application. This could result in less fgraph iterations, but this doesn't mean it will be faster globally. Notes ----- We can put LocalOptimizer and Optimizer as EquilibriumOptimizer suppor both. It is probably not a good idea to have ignore_newtrees=False and tracks_on_change_inputs=True cC`s;tt|ƒjƒ||_||_i|_i|_dS(N(tsuperR[Rtignore_newtreesttracks_on_change_inputst __final__t __cleanup__(R R]R^((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyRs    cO`st|jdtƒ}|jdtƒ}|o-| s7t‚tt|ƒj||||Ž||j|<||j|tsRdRN(R\RqR4R&RRhRMRrR$RR*RtgetattrRR1R%RRFRRtsortRdtseq_optR ( R R!R?RoRMt extra_optRRvRlR"tret((Rxs0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyR4Ms2   %    , cC`s’t|jjdt|ƒd|ƒt|jjƒƒ}d„}|j|ƒtd|d|ƒtd|jd|ƒtd|j d|ƒdS(Ns (id %i)RDcS`s$|d|dk|d|dkS(Ni((tatb((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pytcss positions namess db( RERRRFRNRrtitemsR{RR(R RGt positionsR((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyRH}s$  cC`s tƒ}|j|ƒ|jƒS(N(RRHtgetvalue(R tsio((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyRP‰s  (RRIRYRt SeqOptimizerR|twarnRR#R4RJRKRHRP(((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyRq,s   0 t LocalGroupDBcB`s5eZdZeeejd„Zd„Zd„ZRS(s› Generate a local optimizer of type LocalOptGroup instead of a global optimizer. It supports the tracks, to only get applied to some Op. cC`sDtt|ƒjƒd|_||_||_i|_||_dS(N( R\RˆRRRdtapply_all_optstprofileRrt local_opt(R R‰RŠR‹((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyR˜s     cO`s®tt|ƒj|||Œ|jddƒ}|dkr‚t|jƒdkr_d|j| Generate a Global Optimizer of type TopoOptimizer. t in_to_outcC`s;tt|ƒjƒ||_||_||_||_dS(N(R\RRtdbtorderR]Rd(R RRR]Rd((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyR¿s    cO`s.tj|jj||Ž|j|j|jƒS(N(Rt TopoOptimizerRR4RR]Rd(R R!R?((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyR4ÇsN(RRIRYRRRR4(((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyR¸s tProxyDBcB`s eZdZd„Zd„ZRS(s“ Wrap an existing proxy. This is needed as we can't register the same DB mutiple times in different positions in a SequentialDB. cC`s(t|tƒstdƒ‚||_dS(Nt(RR RR(R R((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyR×scO`s|jj||ŽS(N(RR4(R R!R?((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyR4Ûs(RRIRYRR4(((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyR’Îs (t __future__RRRR%RJt theano.compatRttheano.misc.ordered_setRtsixRRt theano.gofRttheanoRtobjectR R*R[RqRˆRR’(((s0/tmp/pip-build-X4mzal/theano/theano/gof/optdb.pyts  “JCc)