BVhdZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl m Z dd lmZd Zd Zej$fd Zd Zy)z,Autograph specifc overrides for tf.data.ops.N) control_flow) py_builtins) iterator_ops)tensor_conversion) tensor_spec)cond)nestct|tjsJtt dj |t|t ttttjfrtj|}t|dd}||jk7r6|dn t|}t!dj |||j|y)aPVerifies that a symbol has a type compatible vith a given spec. Here, compatibility is viewed in the general TensorFlow sense: that the dtypes are the same after implicit conversion, if both are tensors. This verifier ensures consistent treatment of types across AutoGraph. Args: input_name: A name to use for `input_` in error messages. spec_name: A name to use for `spec` in error messages. input_: Any, value to verify. spec: TypeSpec that `input_` must be compatible with. Raises: ValueError if the two types have been determined not to be compatible. Nz{} cannot be Nonedtypezno dtypez6{} must have the same dtype as {}. Expected {}, got {}) isinstancer TensorSpecinput ValueErrorformatboolintfloatstrnpndarrayrconvert_to_tensor_v2getattrr TypeError) input_name spec_nameinput_spec input_dtypeinput_dtype_strs ]/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/data/ops/iterator_autograph.py_verify_spec_compatibler!s" D+00 11 1 ] (// ; <<sE3 ;<  3 3F ;F.+DJJ$/$7jS=MO @GG  4::  c  tj||dtj tjt||||y#ttf$r+}tdj ||t ||d}~wwxYw)aVerifies that possibly-structured symbol has types compatible vith another. See _verify_spec_compatible for a more concrete meaning of "compatible". Unspec _verify_spec_compatible, which handles singular Tensor-spec objects, verify_structures_compatible can process structures recognized by tf.nest. Args: input_name: A name to use for `input_` in error messages. spec_name: A name to use for `spec` in error messages. input_: Any, value to verify. May, but doesn't need to, be a structure. spec: Any, value that `input_` must be compatible with. May, but doesn't need to, be a structure. Raises: ValueError if the two types have been determined not to be compatible. T)expand_compositesz2{} must have the same element structure as {}. {}N) r assert_same_structurerrrr map_structure functoolspartialr!)rrrres r _verify_structure_compatibler*Ds" vttD/YG   i  >EE  3q6     sA B&BBctjur t|S|j}t dd|j t j |j|jfdS)Nzthe default argumentz the iteratecSN)defaultsr z#_next_tf_iterator..msgr") r UNSPECIFIEDnextget_next_as_optionalr* element_specr has_value get_value)iteratorr/ opt_iterates ` r _next_tf_iteratorr9csk ''' >--/+mWh6K6K {44o r"ctjjtjt t jjtjt jyr-) r next_registryregisterr OwnedIteratorr9rfor_loop_registry_tf_iterator_for_stmtr.r"r register_overridesr@qsI $$  "3  ))  ,"D"Dr")__doc__r'numpyr%tensorflow.python.autograph.operatorsrrtensorflow.python.data.opsrtensorflow.python.frameworkrrtensorflow.python.opsrtensorflow.python.utilr r!r*r1r9r@r.r"r rHsE3>=393&'#L >)4(?(? r"