| 34 | namespace graph |
| 35 | { |
| 36 | void DotGraphVisitor::visit(ActivationLayerNode &n) |
| 37 | { |
| 38 | std::stringstream ss; |
| 39 | ss << n.activation_info().activation(); |
| 40 | _info = ss.str(); |
| 41 | } |
| 42 | |
| 43 | void DotGraphVisitor::visit(BatchNormalizationLayerNode &n) |
| 44 | { |
no test coverage detected