ó àÆ÷Xc@`swdZddlmZmZmZddlZddlZddlmZddlZddl m Z ddl m Z ddl mZejdƒZiejƒd 6ejƒd 6ejƒd 6ejd eƒd 6ejjdeƒd6ejjdeƒd6ejjd edeƒd6ejjd edeƒd6Zd„ZgZej js`dgZnejdgdeƒZejddgdeƒZejddgdeƒZej dƒZ!ejdddgdeƒZ"ejddgdeƒZ#de#_$de_%de_%de_%d e!_%d!e"_%d"e#_%ied6ed#6ed6ed6e!d$6e"d6e#d%6Z'd&„Z(d'ej)fd(„ƒYZ*d)ej)fd*„ƒYZ+d+ej)fd,„ƒYZ,ej-ƒZ.e.j/d-ej0ƒddddƒej.j1d.ed/eƒZ2e.j/d0ej.j3e2d1ej4j5j6ƒd2ddƒe.j/d3ej0ƒd4dddƒe.j/d5ej7d6eƒd7ddd8ƒe.d5j/dej4j0ƒddd9eƒe.j/d:ej0ƒd;dddƒe.j/d<e,d=ƒd>ƒe.j/d%ej7ƒd?dƒe.j/d@e,dAƒdBƒe.j/dCej7ƒdDddƒe.j/dEej7ƒdFdƒe.j/dGej7ƒdHddƒe.j/dIej0ƒdJddƒe.j/dKe*ƒdLddMƒe.j/dNej0ƒdOddƒdPe8fdQ„ƒYZ9e9ejjjdedReƒdƒZ:ej jr/e9ddƒZ;ne9ddƒZ;ie:dS6e;dT6Z<da=dU„Z>dV„Z?dW„Z@dS(Xs WRITEME i(tabsolute_importtprint_functiontdivisionN(tgof(tconfig(t _output_guard(t string_typesstheano.compile.modetpytcsc|pytallow_gcs c|py_nogct use_clooptvmtcvmtvm_nogctcvm_nogccC`s-|tkrtd|ƒ‚n|t|„s tmerge1tapply_all_optstprofiletuselesstfailure_callbackg333333ã?smerge1.1gÍÌÌÌÌÌä?t canonicalizetignore_newtreesitcanonicalize_dbtcleanupsmerge1.2g333333ó?s Print1.21sPost-canonicalizeg\Âõ(\ó?gø?s Print1.51sPost-stabilizeg)\Âõ(ø?t specializeituncanonicalizeitspecialize_devicegÍÌÌÌÌLH@tmerge2i1tadd_destroy_handlergÀH@tinplacetmerge3idtModecB`s‰eZdZddd„Zd„Zd„Zd„Zd„Ze eƒZ d„Z d„Z d „Z d „Zd „Zdd d „ZRS(sQ The Mode represents a way to optimize and then link a computation graph. Parameters ---------- optimizer : a structure of type Optimizer An Optimizer may simplify the math, put similar computations together, improve numerical stability and various other improvements. linker : a structure of type Linker A Linker decides which implementations to use (C or Python, for example) and how to string them together to perform the computation. See Also -------- predefined_linkers predefined_optimizers predefined_modes tdefaultcC`sJ|dkrtj}n|dkr0tj}ntj|||fƒdS(NRV(R"RRt optimizerRUt __setstate__(R0RRW((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pyR=ïs     cC`s|j|jfS(N(tprovided_linkertprovided_optimizer(R0((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pyt __getstate__ÿscC`s½|\}}||_||_t|tƒs9|dkrFt|}n||_t|tƒsj|dkrwt|}nt|tj ƒr•||_n||_ d|_ d|_ ||_ dS(Ni(RYRZt isinstanceRR"RRR%RtQueryt _optimizert call_timetfn_timetmode(R0tstateRRW((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pyRXs          cC`sd|jj|j|jfS(Ns%s(linker = %s, optimizer = %s)(t __class__R8RYRZ(R0((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pyt__str__s cC`s0t|jtjƒr%tj|jƒS|jSdS(N(R\R^RR]toptdbtquery(R0((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pyt__get_optimizerscC`sZt|tƒs|dkr(t|}nt|tƒsC|dkrPt|}n||fS(N(R\RR"RR%(R0RRW((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pytget_linker_optimizers   cG`s7|j|j|jƒ\}}|jd|j|ŒƒS(NRW(RhRYRZtclonet including(R0ttagstlinkR&((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pyRj&s cG`s7|j|j|jƒ\}}|jd|j|ŒƒS(sšAdds new optimization instances to a mode. This method adds new optimization instances to a compilation mode. It works like the `including()` method but takes as inputs optimization instances to add instead of tags. Parameters ---------- optimizations : Every element of `optimizations` is a tuple containing an optimization instance and a floating point value indicating the position at which to insert the optimization in the mode. Returns ------- Mode Copy of the current Mode which includes the provided optimizations. RW(RhRYRZRitregister(R0t optimizationsRlR&((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pyRm-s cG`s7|j|j|jƒ\}}|jd|j|ŒƒS(NRW(RhRYRZRit excluding(R0RkRlR&((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pyRoFs cG`s7|j|j|jƒ\}}|jd|j|ŒƒS(NRW(RhRYRZRit requiring(R0RkRlR&((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pyRpKs tcK`s^|dkri}n|jj|}|dkr?|j}nt|ƒd|d|ƒ}|S(sÇ Create a new instance of this Mode. Keyword arguments can be provided for the linker, in which case its `clone` method will be called with these arguments. RqRRWN(R"RRiRZttype(R0t link_kwargsRWtkwargst new_linkertnew_mode((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pyRiPs     N(R8R9R:R"R=R[RXRdt_Mode__get_optimizertpropertyRWRhRjRmRoRpRi(((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pyRUÚs          tc_thunkst FAST_COMPILEtFAST_RUNcC`sÔ|dkrtj}n|}t|tƒs1|S|dkrtr|tkrbt|jj}n|}tjj|krtSn|d kr |dkr¾ddl m }|dtj ƒ}q4|dkrøddl m }|tttdtj ƒ}q4t|dƒ}n)|tkr$t|}ntd |ƒ‚|dkrÐtjjrm|jtjjjd ƒŒ}ntjjrš|jtjjjd ƒŒ}ntjjrÇ|jtjjjd ƒŒ}n|an|S( NRUt DebugModet NanGuardModei(R|RW(R}s2(linker=config.linker, optimizer=config.optimizer)s'No predefined mode exist for string: %st:(sModes DebugModes NanGuardMode(R"RRaR\Rtinstantiated_default_modetpredefined_modesRcR8t debugmodeR|RWt nanguardmodeR}tTruetevalt ExceptionRDtoptimizer_excludingRotsplittoptimizer_includingRjtoptimizer_requiringRp(t orig_stringtstringtdefault_mode_classR|tretR}((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pytget_modevsB             ! ! ! cC`s tdƒS(N(RŽR"(((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pytget_default_mode«scC`s-|tkrtd|ƒ‚n|t|t SequenceDBReRmtMergeOptimizert LocalGroupDBt local_uselesstTopoDBR&tNavigatorOptimizert warn_inplacet EquilibriumDBtobjectRURzR{R€RRŽRR(((s3/tmp/pip-build-X4mzal/theano/theano/compile/mode.pyts¾                   "           Ž     5