| 496 | |
| 497 | template <typename T> |
| 498 | void printShapes(std::ostream& os, const char* phase, const T& shapes) |
| 499 | { |
| 500 | if (shapes.empty()) |
| 501 | { |
| 502 | os << "Input " << phase << " shapes: model" << std::endl; |
| 503 | } |
| 504 | else |
| 505 | { |
| 506 | for (const auto& s : shapes) |
| 507 | { |
| 508 | os << "Input " << phase << " shape: " << s.first << "=" << s.second << std::endl; |
| 509 | } |
| 510 | } |
| 511 | } |
| 512 | |
| 513 | std::ostream& printBatch(std::ostream& os, int32_t maxBatch) |
| 514 | { |