&VfTdZddlZddlmZddlmZmZmZdZ dZ dZ dZ d Z d Zd Zd Zd ZdZdZdZdZdZdZieje eje ejeejeejeejeejeejeej eej!eej"eej#eej$eej%eej&eej'j(eej)j(eej*j(eej+j(eej,j(eiZ-ej.j(e iZ/dZ0dZ1dZ2dZ3e4ejejejejej"ej!ejej'ej)ej*ej+ej,ejej gZ5GddZ6dZ7dZ8dZ9dZ:dZ;dZiej?e>gej$ez7SlowAppendFloat16ArrayToTensorProto..s!999q  " "999rhalf_valextend tensor_proto proto_valuess r#SlowAppendFloat16ArrayToTensorProtor"s8  99L999rctj|tjjtjSr)r r rbfloat16as_numpy_dtyperrrrs rExtractBitsFromBFloat16r&!s4 1FO:;;; bi rcN|jd|DdS)Nc,g|]}t|Sr)r&rs rrz8SlowAppendBFloat16ArrayToTensorProto..+s!:::  # #:::rrrs r$SlowAppendBFloat16ArrayToTensorProtor))s8  ::\:::rcN|jd|DdS)Nc6g|]}|Srrrs rrz7SlowAppendFloat32ArrayToTensorProto..0 "B"B"B16688"B"B"Br) float_valrrs r#SlowAppendFloat32ArrayToTensorProtor//.!!"B"B\"B"B"BCCCCCrcN|jd|DdS)Nc6g|]}|Srr,rs rrz7SlowAppendFloat64ArrayToTensorProto..4 #C#C#CAFFHH#C#C#Cr) double_valrrs r#SlowAppendFloat64ArrayToTensorProtor53.""#C#Cl#C#C#CDDDDDrcN|jd|DdS)Nc6g|]}|Srr,rs rrz3SlowAppendIntArrayToTensorProto..8s @ @ @a @ @ @rint_valrrs rSlowAppendIntArrayToTensorProtor;7s. @ @< @ @ @AAAAArcN|jd|DdS)Nc6g|]}|Srr,rs rrz5SlowAppendInt64ArrayToTensorProto..<r-r) int64_valrrs r!SlowAppendInt64ArrayToTensorProtor?;r0rcN|jd|DdS)NcBg|]}|dS)rr,rs rrz4SlowAppendQIntArrayToTensorProto..@s$ C C C1 C C Crr9rs r SlowAppendQIntArrayToTensorProtorB?s. C Cl C C CDDDDDrcN|jd|DdS)Nc6g|]}|Srr,rs rrz6SlowAppendUInt32ArrayToTensorProto..Dr3r) uint32_valrrs r"SlowAppendUInt32ArrayToTensorProtorFCr6rcN|jd|DdS)Nc6g|]}|Srr,rs rrz6SlowAppendUInt64ArrayToTensorProto..Hr3r) uint64_valrrs r"SlowAppendUInt64ArrayToTensorProtorJGr6rcN|jd|DdS)NcXg|]'}|j|jfD]}|(Srrealimagrrrvs rrz9SlowAppendComplex64ArrayToTensorProto..M6BBBa!&!&1ABBABBBBr) scomplex_valrrs r%SlowAppendComplex64ArrayToTensorProtorTK8$$BB<BBBrcN|jd|DdS)NcXg|]'}|j|jfD]}|(SrrMrPs rrz:SlowAppendComplex128ArrayToTensorProto..SrRr) dcomplex_valrrs r&SlowAppendComplex128ArrayToTensorProtorYQrUrcN|jd|DdS)Nc6g|]}tj|Srras_bytesrs rrz6SlowAppendObjectArrayToTensorProto..Xs"#M#M#M1FOA$6$6#M#M#Mr) string_valrrs r"SlowAppendObjectArrayToTensorProtor_Ws.""#M#M #M#M#MNNNNNrcN|jd|DdS)Nc6g|]}|Srr,rs rrz4SlowAppendBoolArrayToTensorProto..\s !A!A!Aq!&&((!A!A!Ar)bool_valrrs r SlowAppendBoolArrayToTensorProtorc[s.  !A!AL!A!A!ABBBBBrc|D]\}}||kr|cStD]\}}||kr|cSdSN)items BACKUP_DICT) dtype_dictr keyvals rGetFromNumpyDTypeDictrk|ss$$&&S %<<JJJ %%''S %<<JJJ  4rc|jtjks|jtjkrtSt t |Sre)typer string_unicode_r_rk_NP_TO_APPEND_FNr s rGetNumpyAppendFnrqs8  zRZ5:#<#<11 !15 9 99rct|ttfsgS|sdgSt|gt |dzS)z9Returns the inferred dense dimensions of a list of lists.r) isinstancelisttuplelen_GetDenseDimensions) list_of_listss rrwrwsR mdE] 3 3L Ls M""#&9-:J&K&KKKrc#Kt|ttfr|D]}t|D]}|VdS|VdSre)rsrtru_FlattenToStrings)nested_stringsinnerflattened_strings rrzrzsu.4-00# ' 'E$5e$<$< ' ' &&&&& ' ' 'rceZdZdZdZdS)_Messagec||_dSre_message)selfmessages r__init__z_Message.__init__s  rc|jSrer)rs r__repr__z_Message.__repr__s }rN)__name__ __module__ __qualname__rrrrrrrs2   rrc|D]}||cS dSrer)lrs r _FirstNotNoners)  =HHH  4rc(|tdS|S)NNone)rrQs r_NotNonersyrc^t|ttfs|St|trtd|DsdSt|tr2td|Dst d|DSt d|DS)Nc3NK|] }t|ttfV!dSrersrtrurs r z_FilterTuple..0;;A:a$//;;;;;;rc3NK|] }t|ttfV!dSrerrs rrz_FilterTuple..rrcNg|]"}t|ttfrdn|#Srerrs rrz _FilterTuple..s/HHHAe}55<1HHHrc,g|]}t|Sr) _FilterTuplers rrz _FilterTuple..s555a,q//555r)rsrtruanyrrs rrrs a$ ' '!U;;;;;;; 4!T;;;;;;;  HHaHHH  551555 6 66rct|ttfrtd|DSt|tjt jfrdnt|S)Nc,g|]}t|Sr) _FilterIntrs rrz_FilterInt..777jmm777r) rsrtrurrintegral_typesr Dimensionrrs rrrsf!dE]##977Q777888 a&/1GH I I  a[[rct|ttfrtd|DSt|tjrdnt |S)Nc,g|]}t|Sr) _FilterFloatrs rrz _FilterFloat..s999!l1oo999r)rsrtrurr real_typesrrs rrrsW!dE]##;99q999:::a!233 D44!Drct|ttfrtd|DSt|tjrdnt |S)Nc,g|]}t|Sr)_FilterComplexrs rrz"_FilterComplex..s ;;;AnQ//;;;r)rsrtrurr complex_typesrrs rrrsW!dE]##=;;;;;<<<a!566 G44HQKKGrct|ttfrtd|DSt|tjrdSt |S)Nc,g|]}t|Sr) _FilterStrrs rrz_FilterStr..rr)rsrtrurrbytes_or_text_typesrrs rrrs[!dE]##977Q777888!V/00t{{rct|ttfrtd|DSt|trdnt |S)Nc,g|]}t|Sr) _FilterBoolrs rrz_FilterBool..s888k!nn888r)rsrtrurboolrrs rrrsT!dE]##:88a888999a&& 744HQKK7rc  | |tvrdSt|}tfd|D}|=td|jdt |dt |jddS)Nc&g|] }|Srr)rfnvaluess rrz&_Assertconvertible..s!;;;Rbbjj;;;rz Expected , got z of type 'z ' instead.) _TF_TO_IS_OKgetr TypeErrornamereprrmr)rr fn_listmismatchs` r_Assertconvertiblers }\11u%%G;;;;7;;;<>>>4>>+B+B+B D   rFc " t|tjr|S|rtj|}|tjtjtjtjtj fv}t|tj tj fr"|r| |j}n|}ntt!|dds$tt!|ddt"rtj||}|}nj|t'd|r|jr|j}nd}|;tj|tjdkrtj||}nt1||tj||}t5|jt9|kr:|s8t'd|dt5|jd t9|d |jtjkr"| | tj}nM|jtjkr8|6| tj }tj!||r|}tj|j}|tEd |jz|r|}|=tG|d r|j$|j$krtEd |d|jd|||j}d} |j%} nd|D}tj|tj} | |j%k} |r@|jtM|ks(tEdtM|d|jd |j%| krt'd| |j%fztj|j'tQj)|*} | rL|tVvrC| dkr=|j%|j,zdkrt'd|-| _.| S|tj/krt|tj sjta|} d| D} n1#tD$r$tEdtc|d|dwxYw| j23| | S|4} tk| j}|tEd|j6z|| | | S)aCreate a TensorProto. Args: values: Values to put in the TensorProto. dtype: Optional tensor_pb2 DataType value. shape: List of integers representing the dimensions of tensor. verify_shape: Boolean that enables verification of a shape of values. Returns: A `TensorProto`. Depending on the type, it may contain data in the "tensor_content" attribute, which is not directly useful to Python programs. To access the values you should convert the proto back to a numpy ndarray with `tensor_util.MakeNdarray(proto)`. If `values` is a `TensorProto`, it is immediately returned; `dtype` and `shape` are ignored. Raises: TypeError: if unsupported types are provided. ValueError: if arguments have inappropriate values or if verify_shape is True and shape of values is not equals to a shape from the argument. make_tensor_proto accepts "values" of a python scalar, a python list, a numpy ndarray, or a numpy scalar. If "values" is a python scalar or a python list, make_tensor_proto first convert it to numpy ndarray. If dtype is None, the conversion tries its best to infer the right numpy data type. Otherwise, the resulting numpy array has a convertible data type with the given dtype. In either case above, the numpy ndarray (either the caller provided or the auto converted) must have the convertible type with dtype. make_tensor_proto then converts the numpy array to a tensor proto. If "shape" is None, the resulting tensor proto represents the numpy array precisely. Otherwise, "shape" specifies the tensor's shape and the numpy array can not have more elements than what "shape" specifies. __array__N__array_interface__r zNone values not supported.rz!Argument must be a dense tensor: z - got shape z , but wanted .zUnrecognized data type: %s base_dtypezInconvertible types: z vs. z . Value is Tc,g|]}t|Sr)int)rdims rrz%make_tensor_proto..s+++cS+++rzExpected Tensor's shape: rz>Too many elements provided. Needed at most %d, but received %d)r rlz>Cannot create a tensor proto whose content is larger than 2GB.c6g|]}tj|Srr\rs rrz%make_tensor_proto..s"CCC&/!,,CCCrz!Failed to convert object of type z to Tensor. Contents: z0. Consider casting elements to a supported type.z-Element type not supported in TensorProto: %s)7rsr TensorProtoras_dtypeqint8quint8qint16quint16qint32r ndarraygenericastyper%callablegetattrdictr ValueErroris_numpy_compatibleprodint64emptyrarrayrtshaperwr float64float32int32 array_equalrhasattrrsizeruas_datatype_enumras_shapeas_proto_TENSOR_CONTENT_TYPESitemsizetobytestensor_contentstringrzrmr^rravelrqr)rr r verify_shape is_quantizednparraynp_dtdowncasted_array numpy_dtype is_same_size shape_sizer r! str_values append_fns rmake_tensor_protor$ssV&*011  '&&    L&2:rz2333+  mmE$899GGGG '&+t44 5 5.+-t44d::.+ *V5111 >9:: :  U. (EEE  bh!?!?!?1!D!DhuE222GG vu - - -hvU333GW]##':6'B'BBB$C!jvvtGM22224G4O4O4O4OQ MRZ ' 'U]nnRZ00GGmrx''U]&~~bh77 ~.88 +*/'-00K4w}DEEE   E< ( (  {5 5 5iuugmmmVV -    }  \ ++U+++WU"(333 !W\1  =E%LL00iU||||W]]]4 <* $ $Pw|,-  )*!*511::<<L  '<<<a <'* *w 7 7P '.oo&7&7 #fm##Jvrz,J,J#(00  CClCCCJJ   )&*&\\\\666;   &&z222==??L !344I ;k>N N   IlL))) s ( Q55.R#c d|jjD}tj|tj}t j|j}|j}|j r@tj |j | |S|t j ks|t jkrt|jdkr_tj|jdtj}|j|_tj|| |Stj|jtj}|j|_| |S|t jkrt|jdkrGtjtj|jd|| |Stj|j| |S|t jkrt|jdkrGtjtj|jd|| |Stj|j| |S|t jt jt jt jt jt jt jt jt j t j!f vrt|j"dkrGtjtj|j"d|| |Stj|j"| |S|t jkrt|j#dkrGtjtj|j#d|| |Stj|j#| |S|t j$krt|j%dkrGtjtj|j%d|| |StjtM|j%| |S|t j'krtQ|j)}t|j)dkr`tjtjtU|j)d|j)d|| |StjdtW||D| |S|t j,krtQ|j-}t|j-dkr`tjtjtU|j-d|j-d|| |StjdtW||D| |S|t j.krt|j/dkrGtjtj|j/d|| |Stj|j/| |Stad|jz) aCreate a numpy ndarray from a tensor. Create a numpy ndarray with the same shape and data as the tensor. Args: tensor: A TensorProto. Returns: A numpy array with the tensor contents. Raises: TypeError: if tensor has unsupported type. cg|] }|j Sr)r)rds rrz make_ndarray..s 5 5 5QV 5 5 5rr rrcFg|]}t|d|dSrrcomplexrs rrz make_ndarray..Q*:::1qt$$:::rcFg|]}t|d|dSrrrs rrz make_ndarray.._rrzUnsupported tensor type: %s)1rrr rrrrr r%r frombuffercopyreshaper r$rvrrrrepeatfromiterrr.rr4ruint8int16int8rrrrrr:r>rr^rt complex64iterrSrzip complex128rXrrbr)tensorr num_elements tensor_dtyper tmpits r make_ndarrayrsK 6 5V04 5 5 5E75111L?6<00L  'E cF M&/u = = = TVV WU^^  ' '<6?+J+J v  1 $ $(6?1-RY???C$3CI9S,//77>> >+foRY???C$3CI;;u%% %  ' ' v A % %9)!,E:::Lgenn ;v/u===EEeLL L  ' ' v ! !Q & &9*1-U;;;\genn ;v0>>>FFuMM M              v~  ! # #9*%888,genn ;v~U;;;CCEJJ J  % % v A % %9)!,E:::Lgenn ;v/u===EEeLL L  & & v ! !Q & &9*1-U;;;\genn 8D!2335AAAII%PP P ) ) ) &% & & v" # #q ( (9F/2F4G4JKK  genn  8::c"bkk:::%genn  * * * &% & & v" # #q ( (9F/2F4G4JKK  genn  8::c"bkk:::%genn   $ $ v  1 $ $9+5999<genn ;ve<<<DDUKK K5 DEEEr)NNF)E__doc__numpyr tensorboard.compat.protor"tensorboard.compat.tensorflow_stubrrrrr"r&r)r/r5r;r?rBrFrJrTrYr_rcr rrrrrruint32uint64rrr r object_bool_rr%rrrrrpr$rgrkrqrwrz frozensetrrrrrrrrrrrrrrrrrrrrs\,+//////KKKKKKKKKKBBB  DDDEEEBBBDDDEEEEEEEEE  OOOCCCJ3J3J3H-  H/  H- I.I1I1G ,H-L7M9J2H.  L!A!" M "B#$ M "B N!#C M "B)2 O"$H :::LLL"            ( 7 7 7EEE HHH 888  K+ ' ~& N\N   N\N   N\N  L:, L:, L:, K* MJ - MJ - L:|, NZ. MJ -  MJ