a \=icY ã@s<dZddlmZddlmZddlmZGdd„deƒZ dS)z&A python interface for Grappler items.é)Úop_performance_data_pb2)Úmeta_graph_pb2)Ú_pywrap_tf_itemc@sTeZdZdZddd„Zddd„Zdd „Zd d „Zed d „ƒZ edd„ƒZ dd„Z dS)ÚItemz GrapplerItem.TFcCs:||_t ¡|_|j |¡||_||_d|_| ¡dS)a|Creates an Item. Args: metagraph: a TensorFlow metagraph. ignore_colocation: if set, the tool will ignore all the colocation constraints generated by TensorFlow. ignore_user_placement: if set, all the placement annotations annotated in the metagraph will be ignored. Raises: ValueError: the metagraph is incomplete or invalid. N) Ú _metagraphrZ MetaGraphDefÚ _item_graphÚCopyFromÚ_ignore_colocationÚ_ignore_user_placementÚ_tf_itemÚ _BuildTFItem)ÚselfÚ metagraphZignore_colocationZignore_user_placement©rúp/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/tensorflow/python/grappler/item.pyÚ__init__s  z Item.__init__cCst |j|¡S©N)Útf_itemZTF_IdentifyImportantOps)r Zsort_topologicallyrrrÚIdentifyImportantOps0szItem.IdentifyImportantOpscCsPt |j¡}i}| ¡D]2\}}g}|D]}| tjj |¡¡q(|||<q|S)zGet Op properties.)rZTF_GetOpPropertiesÚitemsÚappendrZOpInfoZTensorPropertiesZ FromString)r ÚpropsÚ propertiesÚkeyÚvaluesÚpropÚvaluerrrÚGetOpProperties3s  ÿ zItem.GetOpPropertiescCs t |j¡S)zÉReturn a list of hard colocation constraints. All the nodes in a colocation tuple must be placed on the same device for the model to work. Returns: A list of colocation tuples. )rZTF_GetColocationGroups©r rrrÚGetColocationGroupsAs zItem.GetColocationGroupscCs|jSr)rrrrrrLszItem.metagraphcCs(|j|jkr"| ¡|j |j¡|jSr)rrr rr rrrrrPs z Item.tf_itemcCst |j ¡|j|j¡|_dSr)rZ TF_NewItemrZSerializeToStringr r r rrrrr Ws þzItem._BuildTFItemN)TF)F) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrÚpropertyrrr rrrrrsý     rN) r#Ztensorflow.core.grappler.costsrZtensorflow.core.protobufrZtensorflow.python.grapplerrrÚobjectrrrrrÚs