def text(tag, text):
plugin_data = SummaryMetadata.PluginData(
plugin_name="text",
content=TextPluginData(version=0).SerializeToString())
smd = SummaryMetadata(plugin_data=plugin_data)
string_val = []
for item in text:
string_val.append(item.encode(encoding="utf_8"))
tensor = TensorProto(
dtype="DT_STRING",
string_val=string_val,
tensor_shape=TensorShapeProto(
dim=[TensorShapeProto.Dim(size=len(text))]),
)
return Summary(value=[Summary.Value(tag=tag, metadata=smd, tensor=tensor)])
def add_variable(v, v_idx):
v_name = parameters.get(v.data, None)
exist = False
if not v_name:
v_name, exist = get_variable_name(v, v_idx)
if not exist:
shape_proto = TensorShapeProto(
dim=[TensorShapeProto.Dim(size=d) for d in v.shape])
if v.parent is None:
inputs = []
else:
inputs = [get_func_name(v.parent)]
# print("Variable: {}:{}".format(v_name, inputs))
nodes.append(
NodeDef(name=v_name.encode(encoding='utf-8'),
op='Variable',
input=inputs,
attr={
'shape': AttrValue(shape=shape_proto),
'dtype': AttrValue(type=DT_FLOAT)
}))
return v_name
def from_graph_def(self, graph_def):
variables = graph_def.variables
parameters = graph_def.parameters
functions = graph_def.functions
inputs = graph_def.inputs
nodes = []
scope = {}
for n, v in parameters.items():
shape_proto = TensorShapeProto(
dim=[TensorShapeProto.Dim(size=d) for d in v.shape])
node = NodeDef(
name=n.encode(encoding='utf-8'),
op='Parameter',
input=[],
attr={
'shape': AttrValue(shape=shape_proto),
'dtype': AttrValue(type=DT_FLOAT)
}
)
nodes.append(node)
scope[n] = node
for n, v in inputs.items():
shape_proto = TensorShapeProto(
dim=[TensorShapeProto.Dim(size=d) for d in v.shape])
nodes.append(NodeDef(
name=n.encode(encoding='utf-8'),
op='Variable',
input=[],
attr={
'shape': AttrValue(shape=shape_proto),
'dtype': AttrValue(type=DT_FLOAT)
}
))
for func_name, func in functions.items():
for o in func['outputs']:
if o in scope:
node = scope[o]
node.input.extend([func_name])
else:
if o in variables:
v = variables[o]
shape_proto = TensorShapeProto(
dim=[TensorShapeProto.Dim(size=d) for d in v.shape])
node = NodeDef(
name=o.encode(encoding='utf-8'),
op='Variable',
input=[func_name],
attr={
'shape': AttrValue(shape=shape_proto),
'dtype': AttrValue(type=DT_FLOAT)
}
)
nodes.append(node)
for i in func['inputs']:
if i in variables:
v = variables[i]
shape_proto = TensorShapeProto(
dim=[TensorShapeProto.Dim(size=d) for d in v.shape])
node = NodeDef(
name=o.encode(encoding='utf-8'),
op='Variable',
input=[],
attr={
'shape': AttrValue(shape=shape_proto),
'dtype': AttrValue(type=DT_FLOAT)
}
)
nodes.append(node)
scope[o] = node
nodes.append(NodeDef(
name=func_name,
op=func['type'],
input=func['inputs'],
attr={"arguments": AttrValue(s='a=1'.encode(encoding='utf-8'))}
))
current_graph = GraphDef(node=nodes, versions=VersionDef(producer=22))
event = event_pb2.Event(
graph_def=current_graph.SerializeToString())
self.file_writer.add_event(event)
def tensor_shape_proto(shape):
"""Creates an object matching
https://github.com/tensorflow/tensorboard/blob/master/tensorboard/compat/proto/tensor_shape.proto
"""
return TensorShapeProto(dim=[TensorShapeProto.Dim(size=d) for d in shape])