BVh(dZddlZddlmZddlmZddlmZddlm Z ddlm Z Gdd e Z Gd d e ZGd d e ZGdde ZGdde ZGddeZGdde ZGdde Ze j*ddZdZy)z)Utilities for using the TensorFlow C API.N) api_def_pb2) op_def_pb2)pywrap_tf_session)compat) tf_contextlibceZdZfdZxZS)AlreadyGarbageCollectedErrorc8tt| |d|dy)Nz of type z9 has already been garbage collected and cannot be called.)superr __init__)selfnameobj_type __class__s V/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/framework/c_api_util.pyr z%AlreadyGarbageCollectedError.__init__s+ &  68*5;<=)__name__ __module__ __qualname__r __classcell__rs@rr r s ==rr cHeZdZdZgdZdZejdZdZ y) UniquePtrzDWrapper around single-ownership C-API objects that handles deletion.)_objdeleterr type_namec`||_||_||_tt ||_yN)rrrstrtyper)r robjrs rr zUniquePtr.__init__*s*DIDI DLc^DNrc#K|j t|j|j|jyw)zYields the managed C-API Object, guaranteeing aliveness. This is a context manager. Inside the context the C-API object is guaranteed to be alive. Raises: AlreadyGarbageCollectedError: if the object is already deleted. N)rr rrr s rgetz UniquePtr.get6s0 yy (DNN CC ))Os<>cR|j}|d|_|j|yyr)rr)r r!s r__del__zUniquePtr.__del__Fs( ))C di ll3rN) rrr__doc__ __slots__r contextlibcontextmanagerr$r&rrrr%s.L6) $  rrc"eZdZdZdgZdZdZy)ScopedTFStatusz/Wrapper around TF_Status that handles deletion.statusc6tj|_yr)c_api TF_NewStatusr.r#s rr zScopedTFStatus.__init__Rs$$&DKrcrt1tj tj|jyyyr)r0TF_DeleteStatusr.r#s rr&zScopedTFStatus.__del__Us/ U22> DKK(?rNrrrr'r(r r&r+rrr-r-Ms7j)')rr-c"eZdZdZdgZdZdZy)ScopedTFImportGraphDefOptions>Wrapper around TF_ImportGraphDefOptions that handles deletion.optionsc6tj|_yr)r0TF_NewImportGraphDefOptionsr8r#s rr z&ScopedTFImportGraphDefOptions.__init__as446DLrcrt1tj tj|jyyyr)r0TF_DeleteImportGraphDefOptionsr8r#s rr&z%ScopedTFImportGraphDefOptions.__del__d0 UAAM **4<<8NrNr4r+rrr6r6\sFk)79rr6c"eZdZdZdgZdZdZy)ScopedTFImportGraphDefResultsr7resultsc||_yr)r@)r r@s rr z&ScopedTFImportGraphDefResults.__init__ps DLrcrt1tj tj|jyyyr)r0TF_DeleteImportGraphDefResultsr@r#s rr&z%ScopedTFImportGraphDefResults.__del__sr=rNr4r+rrr?r?ksFk)9rr?c"eZdZdZfdZxZS)ScopedTFFunctionz1Wrapper around TF_Function that handles deletion.cNtt| ||tjy)N)rr!r)r rEr r0TF_DeleteFunction)r funcrrs rr zScopedTFFunction.__init__}s% D* tU%<%<+>r)rrrr'r rrs@rrErEzs9>>rrEc"eZdZdZdgZdZdZy)ScopedTFBufferz8An internal class to help manage the TF_Buffer lifetime.bufferc^tjtj||_yr)r0TF_NewBufferFromStringras_bytesrK)r buf_strings rr zScopedTFBuffer.__init__s..vz/JKDKrcBtj|jyr)r0TF_DeleteBufferrKr#s rr&zScopedTFBuffer.__del__s $++&rNr4r+rrrJrJs@j)L'rrJc<eZdZdZddgZdZdZdZdZdZ d Z y ) ApiDefMapzWrapper around Tf_ApiDefMap that handles querying and deletion. The OpDef protos are also stored in this class so that they could be queried by op name. _api_def_map _op_per_namectj}tj} |j tj |tj ||_tj|i|_ |jD]}||j|j<y#tj|wxYwr) rOpListr0TF_GetAllOpListParseFromString TF_GetBufferTF_NewApiDefMaprTrQrUopr)r op_def_protobufr\s rr zApiDefMap.__init__s$$&L    !C!""5#5#5c#:;//4d C Doo&#%d & C s >B//Ccrt1tj tj|jyyyr)r0TF_DeleteApiDefMaprTr#s rr&zApiDefMap.__del__s2 U55A t001BrcXtj|j|t|yr)r0TF_ApiDefMapPutrTlen)r texts r put_api_defzApiDefMap.put_api_defs $++T3t9=rc*tj}tj|j|t |} |j tj|tj||S#tj|wxYwr) rApiDefr0TF_ApiDefMapGetrTrcrYrZrQ)r op_name api_def_protor^s r get_api_defzApiDefMap.get_api_defsr&&(M    1 17CL IC!##E$6$6s$;< C  C s $A;;BcZ||jvr|j|Std|d)NzNo op_def found for op name .)rU ValueError)r ris r get_op_defzApiDefMap.get_op_defs6$###   w '' 3G9A> ??rc6|jjSr)rUkeysr#s rop_nameszApiDefMap.op_namess    ! ! ##rN) rrrr'r(r r&rerkrorrr+rrrSrSs2 ~.) &2 >@ $rrSc#K|r)tjtj|}ntj} |tj |y#tj |wxYww)a$Context manager that creates and deletes TF_Buffer. Example usage: with tf_buffer() as buf: # get serialized graph def into buf ... proto_data = c_api.TF_GetBuffer(buf) graph_def.ParseFromString(compat.as_bytes(proto_data)) # buf has been deleted with tf_buffer(some_string) as buf: c_api.TF_SomeFunction(buf) # buf has been deleted Args: data: An optional `bytes`, `str`, or `unicode` object. If not None, the yielded buffer will contain this data. Yields: Created TF_Buffer N)r0rMrrN TF_NewBufferrQ)datar^s r tf_bufferrvsX.  & &vt'< =C    C I #E#sAA7AA7A44A7cJtj}||_||_|S)zReturns a wrapped TF_Output with specified operation and index. Args: c_op: wrapped TF_Operation index: integer Returns: Wrapped TF_Output )r0 TF_Outputoperindex)c_oprzrets r tf_outputr}s# # #(#) *rr)r'r)tensorflow.core.frameworkrrtensorflow.python.clientrr0tensorflow.python.utilrr Exceptionr objectrr-r6r?rErJrSr*rvr}r+rrrs 010?)0=9=%%P )V ) 9F 9 9F 9>y> 'V '.$.$b@  r