| 465 | } |
| 466 | |
| 467 | bool OperatorSupportsFusedActivation(OperatorType type) { |
| 468 | switch (type) { |
| 469 | case OperatorType::kAdd: |
| 470 | case OperatorType::kAveragePool: |
| 471 | case OperatorType::kBatchNormalization: |
| 472 | case OperatorType::kConv: |
| 473 | case OperatorType::kDepthwiseConv: |
| 474 | case OperatorType::kDiv: |
| 475 | case OperatorType::kFullyConnected: |
| 476 | case OperatorType::kL2Pool: |
| 477 | case OperatorType::kMaxPool: |
| 478 | case OperatorType::kMul: |
| 479 | case OperatorType::kSub: |
| 480 | case OperatorType::kSquaredDifference: |
| 481 | return true; |
| 482 | default: |
| 483 | return false; |
| 484 | } |
| 485 | } |
| 486 | |
| 487 | void LogSummary(int log_level, const Model& model) { |
| 488 | VLOG(log_level) << "Operators summary (" << model.operators.size() |