| 415 | } |
| 416 | |
| 417 | Color GetOpColor(const Operator& op) { |
| 418 | if ((op.type == OperatorType::kDepthwiseConv) || |
| 419 | (op.type == OperatorType::kConv) || |
| 420 | (op.type == OperatorType::kFullyConnected) || |
| 421 | (op.type == OperatorType::kFakeQuant)) { |
| 422 | // Give some ops a bolder red |
| 423 | return Color(0xC5, 0x39, 0x29); |
| 424 | } else { |
| 425 | return Color(0xDB, 0x44, 0x37); |
| 426 | } |
| 427 | } |
| 428 | |
| 429 | string GetOpLabel(const Model& model, const Operator& op) { |
| 430 | // Use HTML-like labels (http://www.graphviz.org/doc/info/shapes.html#html) |