Vh`ddlmZddlmZddlmZddlmZdZdZ d dee fd Z y) )Optional)NodeDef) AttrValue)TensorShapeProtoci}|t|jd|d<|/t|}ttj|g|d<|S)a'Create a dict of objects matching a NodeDef's attr field. Follows https://github.com/tensorflow/tensorboard/blob/master/tensorboard/compat/proto/attr_value.proto specifically designed for a NodeDef. The values have been reverse engineered from standard TensorBoard logged data. utf_8encoding)sattr)shape)list_output_shapes)rencodetensor_shape_proto ListValue)dtyper r r shapeprotos T/home/dcms/DCMS/lib/python3.12/site-packages/torch/utils/tensorboard/_proto_graph.pyattr_value_protorsY D} 188W8#=>V  '. !* 0C0C:,0W!X  Kcht|Dcgc]}tj|c}Scc}w)zCreate an object matching a tensor_shape field. Follows https://github.com/tensorflow/tensorboard/blob/master/tensorboard/compat/proto/tensor_shape.proto . )size)dim)rDim) outputsizeds rrrs) z R!!1!5!51!= R SS Rs/Nr c |g}t|ts|g}t|jd||t |||S)zCreate an object matching a NodeDef. Follows https://github.com/tensorflow/tensorboard/blob/master/tensorboard/compat/proto/node_def.proto . rr )nameopinputr ) isinstancerrrr)rr r!rr r attributess r node_protor$ sL } eT "  [['[ *  eZ <  r) UnSpecifiedNNNN) typingr%tensorboard.compat.proto.node_def_pb2r'tensorboard.compat.proto.attr_value_pb2r)tensorboard.compat.proto.tensor_shape_pb2rrrtupler$rrr-sC9=F  T  ! E? r