
    AVhKZ                        d Z ddlZddlmZ ddlmZ ddlmZ ddlm	Z
 ddlmZ ddlmZ dd	lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZm Z m!Z! ddl"m#Z# d4de#e!ejH                  f   fdZ%  ed       ejL                  e%            Z'de#e!ejH                  f   fdZ( ed Z) ed Z*d4de#e!ejH                  f   de#e!e)f   de*de#e!ejH                  f   fdZ+  ed       ejL                  e+            Z,de#e!ejH                  f   de#e!e)f   de*de#e!ejH                  f   fdZ- ed Z.d4de#e!ejH                  f   de#e!e.f   de#e!ej^                  f   fdZ0  ed       ejL                  e0            Z1de#e!ejH                  f   de#e!e.f   de#e!ej^                  f   fd Z2 ed! Z3 ed" Z4d4de#e!ejH                  f   de#e!e3f   d#e#e!e4f   de#e!ejH                  f   fd$Z5  ed%       ejL                  e5            Z6de#e!ejH                  f   de#e!e3f   d#e#e!e4f   de#e!ejH                  f   fd&Z7 ed' Z8 ed( Z9d4de#e!ejH                  f   de#e!e8f   de9de#e!e9f   fd)Z:  ed*       ejL                  e:            Z;de#e!ejH                  f   de#e!e8f   de9de#e!e9f   fd+Z<d4de#e!ejH                  f   de#e!ejz                  f   fd,Z>  ed-       ejL                  e>            Z?de#e!ejH                  f   de#e!ejz                  f   fd.Z@ ed/ ZAd4de#e!ejH                  f   d0eAde#e!eAf   fd1ZB  ed2       ejL                  eB            ZCde#e!ejH                  f   d0eAde#e!eAf   fd3ZDy)5zUPython wrappers around TensorFlow ops.

This file is MACHINE GENERATED! Do not edit.
    N)
pywrap_tfe)context)core)execute)dtypes)annotation_types)op_def_registry)ops)op_def_library)deprecated_endpoints)dispatch)	tf_export)TypeVarListAny)	Annotatedreturnc                 V   t         j                   xs t        j                         }|j                  }|j                  r	 t	        j
                  |d|       }|S t        j                  d|       \  }}}}|dd }t        j                          r&d}|j"                  }	t        j$                  d|	||       |\  }|S # t        j                  $ r }t        j                  ||        Y d}~nd}~wt        j                  $ r Y nw xY w	 t        | |      S # t        j                  $ r Y w xY w)zCreates and returns an empty tensor map.

  handle: an empty tensor map

  Args:
    name: A name for the operation (optional).

  Returns:
    A `Tensor` of type `variant`.
  EmptyTensorMapNnamectx)r    )_contextr   _thread_local_datais_eagerr   TFE_Py_FastPathExecute_core_NotOkStatusException_opsraise_from_not_ok_status_FallbackExceptionempty_tensor_map_eager_fallback_SymbolicException_op_def_library_apply_op_helper_executemust_record_gradientinputsrecord_gradient)
r   _ctxtld_resulte__op_outputs_attrs_inputs_flats
             Q/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/gen_map_ops.pyempty_tensor_mapr5      s"    
			0h..0$#\\11&gn (88t%!QXQK'""$F::L,9('	.' && -
##At,,## 
, ## 
s0    B: :DC((D DD D('D(zraw_ops.EmptyTensorMapc                     g }d }t        j                  dd||||       }t        j                         rt        j                  d|||       |\  }|S )Ns   EmptyTensorMap   r)   attrsr   r   r   )r'   r   r(   r*   )r   r   r3   r2   r-   s        r4   r#   r#   C   sY    ,&.,#)s?'""$,9('	.    ) TV_TensorMapErase_key_dtype_atypes.BFloat16_atypes.Bool_atypes.Complex128_atypes.Complex64_atypes.Float16_atypes.Float32_atypes.Float64_atypes.Float8e4m3b11fnuz_atypes.Float8e4m3fn_atypes.Float8e4m3fnuz_atypes.Float8e5m2_atypes.Float8e5m2fnuz_atypes.Half_atypes.Int16_atypes.Int32_atypes.Int4_atypes.Int64_atypes.Int8_atypes.QInt16_atypes.QInt32_atypes.QInt8_atypes.QUInt16_atypes.QUInt8_atypes.Resource_atypes.String_atypes.UInt16_atypes.UInt32_atypes.UInt4_atypes.UInt64_atypes.UInt8_atypes.Variant) TV_TensorMapErase_value_dtyper<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   input_handlekeyvalue_dtypec           	         t         j                   xs t        j                         }|j                  }|j                  r	 t	        j
                  |d|| |d|      }|S t        j                  |d      }t        j                   d| |||      \  }}}	}
|
dd }t        j"                         rHd|	j%                  d      d|	j%                  d      f}|	j&                  }t        j(                  d|||       |\  }|S # t        j                  $ r }t        j                  ||       Y d}~nd}~wt        j                  $ r Y nw xY w	 t        | ||||      S # t        j                  $ r Y w xY w)a  Returns a tensor map with item from given key erased.

  input_handle: the original map
  output_handle: the map with value from given key removed
  key: the key of the value to be erased

  Args:
    input_handle: A `Tensor` of type `variant`.
    key: A `Tensor`.
    value_dtype: A `tf.DType`.
    name: A name for the operation (optional).

  Returns:
    A `Tensor` of type `variant`.
  TensorMapEraser^   Nr^   r   r   r\   r]   r^   r   	key_dtype)r   r   r   r   r   r   r   r   r    r!   r"   tensor_map_erase_eager_fallbackr$   r'   	make_typer%   r&   r(   _get_attr_typer)   r*   r\   r]   r^   r   r+   r,   r-   r.   r/   r0   r1   r2   r3   s                r4   tensor_map_eraserh   R   sr     
			0h..0$#\\11lCg n "";>+'88|&1>!QX QK'""$3--k:M  /1F::L,9('	.- && -
##At,,## 
,
4TK K## 
0    C9 9E D''E ?E E E+*E+zraw_ops.TensorMapErasec                 ^   t        j                  |d      }t        j                  |g|g       \  }\  }t        j                  | t
        j                        } | |g}d|d|f}t        j                  dd||||      }t        j                         rt        j                  d|||       |\  }|S )Nr^   rc   s   TensorMapEraser7   r8   r`   
r'   re   args_to_matching_eagerr    convert_to_tensor_dtypesvariantr   r(   r*   	r\   r]   r^   r   r   _attr_key_dtyper3   r2   r-   s	            r4   rd   rd      s    "";>+$;;SE3K/6C''gooF,$,-E&.,#)s?'""$,9('	.r:   ) TV_TensorMapHasKey_key_dtyper<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   c                    t         j                   xs t        j                         }|j                  }|j                  r	 t	        j
                  |d|| |      }|S t        j                  d| ||      \  }}}}	|	dd }t        j                          r7d|j#                  d      f}
|j$                  }t        j&                  d||
|       |\  }|S # t        j                  $ r }t        j                  ||       Y d}~nd}~wt        j                  $ r Y nw xY w	 t        | |||      S # t        j                  $ r Y w xY w)aJ  Returns whether the given key exists in the map.

  input_handle: the input map
  key: the key to check
  has_key: whether the key is already in the map or not

  Args:
    input_handle: A `Tensor` of type `variant`.
    key: A `Tensor`.
    name: A name for the operation (optional).

  Returns:
    A `Tensor` of type `bool`.
  TensorMapHasKeyNr   )r\   r]   r   rc   )r   r   r   r   r   r   r   r   r    r!   r"   !tensor_map_has_key_eager_fallbackr$   r%   r&   r'   r(   rf   r)   r*   )r\   r]   r   r+   r,   r-   r.   r/   r0   r1   r2   r3   s               r4   tensor_map_has_keyrv      s?    
			0h..0$#\\11|S:gn (88#DJ!QXQK'""$3--k:;F::L<:('	.' && -
##At,,## 
.
$D2 2## 
s0    C D"C==DDD) )D?>D?zraw_ops.TensorMapHasKeyc                 .   t        j                  |g|g       \  }\  }t        j                  | t        j
                        } | |g}d|f}t        j                  dd||||      }t        j                         rt        j                  d|||       |\  }|S )Nrc   s   TensorMapHasKeyr7   r8   rt   	r'   rl   r    rm   rn   ro   r   r(   r*   )r\   r]   r   r   rq   r3   r2   r-   s           r4   ru   ru      s    $;;SE3K/6C''gooF,$,)&/<#)s?'""$<:('	.r:   ) TV_TensorMapInsert_key_dtyper<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   ) TV_TensorMapInsert_value_dtyper<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   valuec                    t         j                   xs t        j                         }|j                  }|j                  r	 t	        j
                  |d|| ||      }|S t        j                  d| |||      \  }}}	}
|
dd }t        j                          rHd|	j#                  d      d|	j#                  d      f}|	j$                  }t        j&                  d|||       |\  }|S # t        j                  $ r }t        j                  ||       Y d}~nd}~wt        j                  $ r Y nw xY w	 t        | ||||      S # t        j                  $ r Y w xY w)a  Returns a map that is the 'input_handle' with the given key-value pair inserted.

  input_handle: the original map
  output_handle: the map with key and value inserted
  key: the key to be inserted
  value: the value to be inserted

  Args:
    input_handle: A `Tensor` of type `variant`.
    key: A `Tensor`.
    value: A `Tensor`.
    name: A name for the operation (optional).

  Returns:
    A `Tensor` of type `variant`.
  TensorMapInsertNr   )r\   r]   r{   r   rc   r^   )r   r   r   r   r   r   r   r   r    r!   r"    tensor_map_insert_eager_fallbackr$   r%   r&   r'   r(   rf   r)   r*   )r\   r]   r{   r   r+   r,   r-   r.   r/   r0   r1   r2   r3   s                r4   tensor_map_insertr      sZ   " 
			0h..0$#\\11|S%Agn (88#U $&!QX QK'""$3--k:M  /1F::L<:('	.+ && -
##At,,## 
-
U49 9## 
s0    C" "D)5DD)(D)-D= =EEzraw_ops.TensorMapInsertc                 n   t        j                  |g|g       \  }\  }t        j                  |g|g       \  }\  }t        j                  | t        j
                        } | ||g}d|d|f}t        j                  dd||||      }	t        j                         rt        j                  d|||	       |	\  }	|	S )Nrc   r^   s   TensorMapInsertr7   r8   r}   rx   )
r\   r]   r{   r   r   rq   _attr_value_dtyper3   r2   r-   s
             r4   r~   r~     s    $;;SE3K/6C ( ? ?b QXe''gooF,U+,-9JK&/<#)s?'""$<:('	.r:   ) TV_TensorMapLookup_key_dtyper<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   ) TV_TensorMapLookup_value_dtyper<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   c           	         t         j                   xs t        j                         }|j                  }|j                  r	 t	        j
                  |d|| |d|      }|S t        j                  |d      }t        j                   d| |||      \  }}}	}
|
dd }t        j"                         rHd|	j%                  d      d|	j%                  d      f}|	j&                  }t        j(                  d|||       |\  }|S # t        j                  $ r }t        j                  ||       Y d}~nd}~wt        j                  $ r Y nw xY w	 t        | ||||      S # t        j                  $ r Y w xY w)an  Returns the value from a given key in a tensor map.

  input_handle: the input map
  key: the key to be looked up
  value: the value found from the given key

  Args:
    input_handle: A `Tensor` of type `variant`.
    key: A `Tensor`.
    value_dtype: A `tf.DType`.
    name: A name for the operation (optional).

  Returns:
    A `Tensor` of type `value_dtype`.
  TensorMapLookupr^   Nra   rb   rc   )r   r   r   r   r   r   r   r   r    r!   r"    tensor_map_lookup_eager_fallbackr$   r'   re   r%   r&   r(   rf   r)   r*   rg   s                r4   tensor_map_lookupr     sr     
			0h..0$#\\11|S-g n "";>+'88#'2?!QX QK'""$3--k:M  /1F::L<:('	.- && -
##At,,## 
-
4TK K## 
ri   zraw_ops.TensorMapLookupc                 ^   t        j                  |d      }t        j                  |g|g       \  }\  }t        j                  | t
        j                        } | |g}d|d|f}t        j                  dd||||      }t        j                         rt        j                  d|||       |\  }|S )Nr^   rc   s   TensorMapLookupr7   r8   r   rk   rp   s	            r4   r   r   L  s    "";>+$;;SE3K/6C''gooF,$,-E&/<#)s?'""$<:('	.r:   c                 \   t         j                   xs t        j                         }|j                  }|j                  r	 t	        j
                  |d||       }|S t        j                  d| |      \  }}}}|dd }t        j                          r&d}	|j"                  }
t        j$                  d|
|	|       |\  }|S # t        j                  $ r }t        j                  ||       Y d}~nd}~wt        j                  $ r Y nw xY w	 t        | ||      S # t        j                  $ r Y w xY w)a  Returns the number of tensors in the input tensor map.

  input_handle: the input map
  size: the number of tensors in the map

  Args:
    input_handle: A `Tensor` of type `variant`.
    name: A name for the operation (optional).

  Returns:
    A `Tensor` of type `int32`.
  TensorMapSizeNr   )r\   r   r   )r   r   r   r   r   r   r   r   r    r!   r"   tensor_map_size_eager_fallbackr$   r%   r&   r'   r(   r)   r*   )r\   r   r+   r,   r-   r.   r/   r0   r1   r2   r3   s              r4   tensor_map_sizer   [  s'    
			0h..0$#\\11ot\3gn (88l?!QXQK'""$F::Lvw8('	.' && -
##At,,## 
+
Tt- -## 
s0    B< <DC**DDD D+*D+zraw_ops.TensorMapSizec                     t        j                  | t        j                        } | g}d }t	        j
                  dd||||      }t	        j                         rt	        j                  d|||       |\  }|S )Ns   TensorMapSizer7   r8   r   )r    rm   rn   ro   r'   r   r(   r*   )r\   r   r   r3   r2   r-   s         r4   r   r     sq    ''gooF,,&-q#)s?'""$vw8('	.r:   ) TV_TensorMapStackKeys_key_dtyper<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   rc   c                    t         j                   xs t        j                         }|j                  }|j                  r	 t	        j
                  |d|| d|      }|S t        j                  |d      }t        j                   d| ||      \  }}}}	|	dd }t        j"                         r7d|j%                  d      f}
|j&                  }t        j(                  d||
|       |\  }|S # t        j                  $ r }t        j                  ||       Y d}~nd}~wt        j                  $ r Y nw xY w	 t        | |||      S # t        j                  $ r Y w xY w)a=  Returns a Tensor stack of all keys in a tensor map.

  input_handle: the input map
  keys: the returned Tensor of all keys in the map

  Args:
    input_handle: A `Tensor` of type `variant`.
    key_dtype: A `tf.DType`.
    name: A name for the operation (optional).

  Returns:
    A `Tensor` of type `key_dtype`.
  TensorMapStackKeysrc   N)rc   r   r   )r\   rc   r   )r   r   r   r   r   r   r   r   r    r!   r"   $tensor_map_stack_keys_eager_fallbackr$   r'   re   r%   r&   r(   rf   r)   r*   )r\   rc   r   r+   r,   r-   r.   r/   r0   r1   r2   r3   s               r4   tensor_map_stack_keysr     sZ    
			0h..0$#\\11"D,g n   K8)'88<9#')!QX QK'""$3--k:;F::LlFG=('	.+ && -
##At,,## 
1
)$DB B## 
s0    C& &D-9DD-,D-1E   EEzraw_ops.TensorMapStackKeysc                    t        j                  |d      }t        j                  | t        j
                        } | g}d|f}t        j                  dd||||      }t        j                         rt        j                  d|||       |\  }|S )Nrc   s   TensorMapStackKeysr7   r8   r   )	r'   re   r    rm   rn   ro   r   r(   r*   )r\   rc   r   r   r3   r2   r-   s          r4   r   r     s      K8)''gooF,,#&2Al#)s?'""$lFG=('	.r:   )N)E__doc__collectionstensorflow.pythonr   tensorflow.python.eagerr   r   r   r   r   r'   tensorflow.python.frameworkr   rn   tensorflow.security.fuzzing.pyr   _atypesr	   _op_def_registryr
   r    r   r%   "tensorflow.python.util.deprecationr   tensorflow.python.utilr   	_dispatch tensorflow.python.util.tf_exportr   typingr   r   r   typing_extensionsr   Variantr5   	to_raw_opr   r#   r;   r[   rh   r`   rd   rr   Boolrv   rt   ru   ry   rz   r   r}   r~   r   r   r   r   r   Int32r   r   r   r   r   r   r   r   r:   r4   <module>r      sq  
  6 7 1 7 9 F K 3 I C 8 6 % % '%9S'//-A#B %N 534^T^^DT5UV	)C<P2Q 	 &  Z
  '  !^
 .9S'//-A#B .SVXsSsIt .  Da .  qz  {~  @G  @O  @O  {O  qP .` 534^T^^DT5UV)C<P2Q Xabe  hC  cC  YD   Sp   @I  JM  OV  O^  O^  J^  @_   '   \
 )YsGOO/C%D )9UXZvUvKw )  HQ  RU  W^  Wc  Wc  Rc  Hd )V 7)56~t~~FX7YZIc7??>R4S Zcdg  jF  eF  [G   W`  ad  fm  fr  fr  ar  Ws   '   \
 !(  "`
 -Ic7??.B$C -)TWYuTuJv -  @I  JM  Om  Jm  @n -  ~G  HK  MT  M\  M\  H\  ~] -^ 7)56~t~~FW7XY9S'//=Q3R Ybcf  iE  dE  ZF   OX  Y\  ^|  Y|  O}   MV  WZ  \c  \k  \k  Wk  Ml   '   \
 !(  "`
 .Ic7??.B$C .)TWYuTuJv .  Fd .  t}  ~A  Ca  ~a  tb .` 7)56~t~~FW7XY9S'//=Q3R Ybcf  iE  dE  ZF   Us   CL  MP  Rp  Mp  Cq ')C,@"A 'QZ[^`g`m`m[mQn 'R 3	12>4>>/3RS
3;O1P 
`ijmovo|o|j|`} 
 #*  #b
 +	#w2F(G +Ts +  DM  NQ  Sr  Nr  Ds +Z =Y;<^T^^La=bc ygooAU7V   dC   S\  ]`  bA  ]A  SB r:   