AVhdZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd Zdd Zej d dZej ddZy )z6Operations for ExtensionTypes (aka Composite Tensors).)composite_tensor_variant_pb2)composite_tensor)dtypes)ops)tensor)gen_composite_tensor_ops)nested_structure_coder)nestNct|tjstdt |d| |j }|j |std|d|dtj}|jjtj|jtj t#j$|d|j'|S)a~Encodes `value` as a scalar variant tensor. Args: value: The `ExtensionType` value to encode. type_spec: Information about the value's type that should be included in the encoding. name: Optional name for the operation. Returns: A Tensor with shape=`()` and dtype=`tf.variant`. Raises: ValueError: If `type_spec` is not compatible with `value`. z3Expected `value` to be a CompositeTensor. Received .z `type_spec` z is not compatible with `value` Texpand_composites) componentsmetadataname) isinstancerCompositeTensor TypeErrortype _type_specis_compatible_with ValueErrorrCompositeTensorVariantMetadatatype_spec_protoCopyFromr encode_structuretype_spec_valuer$CompositeTensorVariantFromComponentsr flattenSerializeToString)value type_specrrs Z/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/ops/composite_tensor_ops.pycomposite_tensor_to_variantsr$s E+;; <  $U }A/ 00  I  % %e , |I;.Ny# $$ ) H H J( ##--i8HHJ " F Fet<))+  c^t|tjstd|d|jt j k7rtd|d|jjdtj}|jjtj|jt!j"|dDcgc]}|j}}t%j&||j)||}t!j*||dScc}w)aReturns the `ExtensionType` value encoded by a variant scalar tensor. Args: encoded: A Tensor returned by `composite_tensor_to_variants`. type_spec: The `TypeSpec` of the original value. This is used to determine the number and types of the component tensors that comprise the decoded value. Must be compatible with the `TypeSpec` serilized in `encoded`. name: Optional name for the operation. Returns: An `ExtensionType` value that is compatible with `TypeSpec`. Raises: TypeError: If `encoded` is not a Tensor with dtype=variant. InvalidArgumentError: If `encoded` is not compatible with `type_spec`. z'Expected `encoded` to be a Tensor, got r z.Expected `encoded` to have dtype=variant, got Tr )encodedr Tcomponentsr)rrTensorrdtypervariantshapeassert_is_compatible_withrrrrr rrr rr"CompositeTensorVariantToComponentsr pack_sequence_as)r(r"rrtcomponent_dtypesrs r#composite_tensor_from_variantr3=s " GV]] + =g[J KK ]]fnn$ Dk$ %% --))"- ) H H J( ##--i8HHJ||IFagg(JJ))+"  *   y* MMsD*rcntj||jd|jdS)Nrr))r(rr))rr/get_attr)opgrads r#"_composite_tensor_to_variants_gradr8es0 ! D D{{:&++m, ..r%r/ct|t|jk(sJtt|Dcgc]}|||j|n||}}tj||j dScc}w)Nr)rr)lenoutputsrangerrr5)r6r7irs r##_composite_tensor_from_variant_gradr>ms Tc"**o %% %>C3t9=M89tAwbjjmDG3* " F Fbkk*&= ??s B)NN)N)__doc__tensorflow.core.protobufrtensorflow.python.frameworkrrrrtensorflow.python.opsrtensorflow.python.saved_modelr tensorflow.python.utilr r$r3RegisterGradientr8r>r'r%r#rFss=A8.+.:@'D%NP<=.>.:;?<?r%