§
    Vp«f
  ã                   óî   — d dl mZmZmZmZmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8 dS )é    )8ÚArrayÚAxisSizeÚ
BatchTraceÚBatchTracerÚBatchingRuleÚ
RaggedAxisÚEltÚFromEltHandlerÚGetIdxÚIndexedAxisSizeÚMakeIotaHandlerÚMapSpecÚ	NotMappedÚJumbleÚ
JumbleAxisÚJumbleTyÚToEltHandlerÚ	VmappableÚZeroÚZeroIfMappedÚaxis_primitive_batchersÚbatchÚbatch_custom_jvp_subtraceÚbatch_custom_vjp_bwdÚbatch_jaxprÚbatch_jaxpr2Úbatch_jaxpr_axesÚbatch_subtraceÚbdim_at_frontÚ	broadcastÚbroadcast_batcherÚdefbroadcastingÚ
defreducerÚdefvectorizedÚflatten_fun_for_vmapÚfrom_eltÚfrom_elt_handlersÚis_vmappableÚ	make_iotaÚmake_iota_handlersÚ	matchaxisÚmoveaxisÚ
not_mappedÚjumble_axisÚprimitive_batchersÚreducer_batcherÚregister_vmappableÚ
spec_typesÚspmd_axis_primitive_batchersÚto_eltÚto_elt_handlersÚunregister_vmappableÚvectorized_batcherÚ
vmappablesÚvtileÚzero_if_mappedN)9Újax._src.interpreters.batchingr   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   © ó    úY/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/interpreters/batching.pyú<module>r?      sS  ðð$9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9ð 9r=   