U <cn@sXddlmZddlmZddlmZddlmZddZddZ dee d d dZ d S))Optional)NodeDef) AttrValueTensorShapeProtocCsNi}|dk r"t|jddd|d<|dk rJt|}ttj|gdd|d<|S) aCreates a dict of objects matching 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. Nutf_8encoding)sattrshape)listZ_output_shapes)rencodetensor_shape_protoZ ListValue)dtyper r r Z shapeprotorH/tmp/pip-unpacked-wheel-gikjz4vx/torch/utils/tensorboard/_proto_graph.pyattr_value_protosrcCstdd|DdS)zCreates an object matching https://github.com/tensorflow/tensorboard/blob/master/tensorboard/compat/proto/tensor_shape.proto cSsg|]}tj|dqS))size)rZDim).0drrr sz&tensor_shape_proto..)Zdimr) outputsizerrrrsr UnSpecifiedNr cCs<|dkr g}t|ts|g}t|jdd||t|||dS)zCreates an object matching https://github.com/tensorflow/tensorboard/blob/master/tensorboard/compat/proto/node_def.proto Nrr)nameopinputr ) isinstancerrrr)rrrrr r attributesrrr node_protos    r!)rNNNNr) typingrZ%tensorboard.compat.proto.node_def_pb2rZ'tensorboard.compat.proto.attr_value_pb2rZ)tensorboard.compat.proto.tensor_shape_pb2rrrtupler!rrrrs