o ?ep@sdZddlZddlmZddlmZddlmZddlmZddlmZddlm Z Gd d d e d d Z d d Z ddZ ddZddZddZddZddZddZddZddZd d!ZdS)"z)Operators specific to slicing operations.N)dtypes) tensor_util) gen_array_ops)gen_string_ops)list_ops)tensor_array_opsc@s eZdZdS) GetItemOptsN)__name__ __module__ __qualname__r r m/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/autograph/operators/slices.pyrsr element_dtypecCs|t|tsJt|tjrt||St|r9|jtj kr#t |||S|jtj kr4|j j dkr4t||St||St||S)aThe slice read operator (i.e. __getitem__). Note: it is unspecified whether target will be mutated or not. In general, if target is mutable (like Python lists), it will be mutated. Args: target: An entity that supports getitem semantics. i: Index to read from. opts: A GetItemOpts object. Returns: The read element. Raises: ValueError: if target is not of a supported type. r) isinstancerr TensorArray_tf_tensorarray_get_itemr is_tf_typedtypervariant_tf_tensor_list_get_itemstringshapeZndims_tf_tensor_string_get_item_tf_tensor_get_item _py_get_item)targetioptsr r r get_item"s        rcCs ||S)z4Overload of get_item that stages a TensorArray read.)readrrr r r rBs rcCs(|jdur tdtj|||jd}|S)z4Overload of get_item that stages a Tensor list read.Nzacannot retrieve from a list without knowing its element type; use set_element_type to annotate itr)r ValueErrorrZtensor_list_get_item)rrrxr r r rGs rcC||S)zAOverload of get_item that stages a Tensor (not Tensor list) read.r r!r r r rPrcCst||d}|S)z6Overload of get_item that stages a Tensor string read.)rsubstrrrr#r r r rUsrcCr$)z>Overload of get_item that executes a Python list modification.r r!r r r r%rcCsRt|tjr t|||St|r#|jtjkrt |||St |||St |||S)aThe slice write operator (i.e. __setitem__). Note: it is unspecified whether target will be mutated or not. In general, if target is mutable (like Python lists), it will be mutated. Args: target: An entity that supports setitem semantics. i: Index to modify. x: The new element value. Returns: Same as target, after the update was performed. Raises: ValueError: if target is not of a supported type. ) rrr_tf_tensorarray_set_itemrrrrr_tf_tensor_list_set_item_tf_tensor_set_item _py_set_itemr(r r r set_item`s       r-cCs |||S)z5Overload of set_item that stages a TensorArray write.)writer(r r r r)|s r)cCst|||S)z6Overload of set_item that stages a Tensor list update.)rZtensor_list_set_itemr(r r r r*sr*cCst||ff|fS)z9Overload of set_item that stages a Tensor scatter update.)rZtensor_scatter_updater(r r r r+sr+cCs |||<|S)z>Overload of set_item that executes a Python list modification.r r(r r r r,sr,)__doc__ collectionsZtensorflow.python.frameworkrrZtensorflow.python.opsrrrr namedtuplerrrrrrrr-r)r*r+r,r r r r s(