BVh 6dZddlmZddlmZddlmZ ddZy)z4Experimental API for optimizing `tf.data` pipelines.)ops) tensor_shape)gen_dataset_opsct|tstd|dt|tstd|dt|tstd|d|j}|Dcgc]}t j |}}|Dcgc]}t j|}}tj||j||||Scc}wcc}w)aMap a function on the list of tensors unpacked from `elems` on dimension 0. Args: fn: A function (`function.defun`) that takes a list of tensors and returns another list of tensors. The output list has the same types as output_dtypes. The elements of the output list have the same dimension 0 as `elems`, and the remaining dimensions correspond to those of `fn_output_shapes`. elems: A list of tensors. output_dtypes: A list of dtypes corresponding to the output types of the function. output_shapes: A list of `TensorShape`s corresponding to the output shapes from each invocation of the function on slices of inputs. max_intra_op_parallelism: An integer. If positive, sets the max parallelism limit of each function call to this. Raises: ValueError: if any of the inputs are malformed. Returns: A list of `Tensor` objects with the same types as `output_dtypes`. z+`elems` must be a list of tensors, but was .z>`output_dtypes` must be a list of `tf.DType` objects, but was zD`output_shapes` must be a list of `tf.TensorShape` objects, but was ) isinstancelist ValueErrorget_concrete_functionrconvert_to_tensorr TensorShaper map_defuncaptured_inputs)fnelems output_dtypes output_shapesmax_intra_op_parallelism concrete_fness a/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/data/experimental/ops/map_defun.pyrrs6 E4 B5'K LL M4 (  -a1 22 M4 ( ))6q: ;;((*+.3 33  # 3% 38EF1<++A.F-F  " "5+*E*E#0-#; == 4Fs 2CCN))__doc__tensorflow.python.frameworkrrtensorflow.python.opsrrrrs;+41() +=r