o ?eR%@s2dZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z dd l Td d Z e d e e de e dddZe dddZe dddZe dee dddZe dddZe dd d!Ze d"d#d$Ze d%d&d'Zd(S))z7Gradients for operators defined in control_flow_ops.py.)dtypes)indexed_slices)ops) sparse_tensor)tensor)control_flow_ops)control_flow_util)math_ops)*c GsJt}|}|}t|trN|jj|}|dur.|ddur,tj ||ddddS|ddurLt |d|dgddd}||jj|<|dfSdSt|t r|d|j }|dury|j djtjkrwt ||j gd d dddfSdSt |d dddfSt|d|j dd}t|d|j dd}t ||gddfS) aGradients for a Switch op is calculated using a Merge op. If the switch is a loop switch, it will be visited twice. We create the merge on the first visit, and update the other input of the merge on the second visit. A next_iteration is also added on second visit. NF)Zenforce_shape_invariant)NNrZb_switchnameZcond_resource_grad cond_grad)rget_default_graph_get_control_flow_context isinstance WhileContext grad_stateZ switch_mapgetrZ_AddNextAndBackEdgemerge CondContextbranchinputsZdtyperresourceswitch) opgradgraphop_ctxt grad_ctxtZ merge_gradZ zero_gradZ false_gradZ true_gradr!h/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/ops/control_flow_grad.py _SwitchGrad s@      r#ZSwitchZ RefSwitchZMergec sjdj}t}t|}|}t|trt |j St|t r`|j }|rX|jrX|j}|j|j} | durV|j}|||} ||| |} | |j|j<| }t j |ddStj} fddt| Dfddt| DS)z:Gradients for a Merge op are calculated using a Switch op.rNrr csg|] }tjd|qSr )r equalZoutputs.0i)rr!r" sz_MergeGrad..cs g|] }t|dqSr$)r_SwitchRefOrTensorr&)condrr!r"r)s)rrrrrZGetOutputContextrrrrr*ZpivotrpredrZ history_maprr Z grad_contextExitZAddForwardAccumulatorEnterZAddBackpropAccumulatedValuelenrange) rr_Zinput_oprrr r,rZ real_predZ history_predZ num_inputsr!)r+rrr" _MergeGrad\s2          r2ZRefMergecCs t|||SN)r2)rrr1r!r!r" _RefMergeGrads r4r-cCst}|}|}|jsdS|jrtdt|tjr%| |j n.t|t j t jfs8tdt|d| |jj | |jj |j}|durS| |j |tj||j d|jdd}|j|||S)z:Gradients for an exit op are calculated using an Enter op.Nz4Second-order gradient for while loops not supported.Type zO not supported, must be either`indexed_slices.IndexedSlices` or `SparseTensor`.FZb_exit) is_constantparallel_iterationsr )rrr back_propr TypeErrorrrTensorZAddNamer r IndexedSlicesrZ SparseTensortypevaluesindices dense_shaper.rZ_Enterr7Z loop_entersappendr-)rrrrr r?resultr!r!r" _ExitGrads6    rBZRefExitZ NextIterationcCs|S)zA forward next_iteration is translated into a backprop identity. Note that the backprop next_iteration is added in switch grad. r!r1rr!r!r"_NextIterationGradsrDZRefNextIterationcC t||Sr3)rDrCr!r!r"_RefNextIterationGrad rFr.cCst}|}|dur|S|js|S|jdur|S|drEt|tjr-| ||}|St|t j r;| ||}|St dt|dt|}|j|||g|S)zGradients for an Enter are calculated using an Exit op. For loop variables, grad is the gradient so just add an exit. For loop invariants, we need to add an accumulator loop. Nr6r5z/ not supported,must be Tensor or Indexed Slices)rrrr8rZget_attrrrr:ZAddBackpropAccumulatorrr;Z#AddBackpropIndexedSlicesAccumulatorr9r<exitZ loop_exitsr@Z ExitResult)rrrr rAr!r!r" _EnterGrads(        rIZRefEntercCrEr3)rI)rrr!r!r" _RefEnterGradrGrJZLoopCondcCsdS)z0Stop backprop for the predicate of a while loop.Nr!)r1r!r!r" _LoopCondGradsrKN)__doc__Ztensorflow.python.frameworkrrrrrZtensorflow.python.opsrrr Z&tensorflow.python.ops.control_flow_opsr#ZRegisterGradientr2r4rBrDrFrIrJrKr!r!r!r"s<        8 /  (   #