draw_graph/
draw_graph.rs

1use tensorboard_rs::summary_writer::SummaryWriter;
2//use tensorboard_proto::event::{Event, TaggedRunMetadata};
3//use tensorboard_proto::summary::{Summary};
4//use tensorboard_proto::graph::{GraphDef, };
5use tensorboard_proto::node_def::{NodeDef, };
6//use tensorboard_proto::versions::{VersionDef, };
7use tensorboard_proto::attr_value::{AttrValue, };
8//use tensorboard_proto::tensor_shape::{TensorShapeProto, };
9//use tensorboard_proto::step_stats::{RunMetadata, };
10use protobuf::RepeatedField;
11use std::collections::HashMap;
12
13pub fn main() {
14    let mut writer = SummaryWriter::new(&("./logdir".to_string()));
15
16    let mut node1 = NodeDef::new();
17    node1.set_name("node1".to_string());
18    node1.set_op("op1".to_string());
19    
20    let inputs = RepeatedField::from(vec![]);
21    node1.set_input(inputs);
22    
23    let mut attrs = HashMap::new();
24    let mut v1 = AttrValue::new();
25    v1.set_i(16);
26    attrs.insert("attr1".to_string(), v1);
27    node1.set_attr(attrs);
28
29    writer.add_graph(&[node1]);
30    
31    writer.flush();
32}
draw_graph/draw_graph.rs

draw_graph/
draw_graph.rs
