| 14 | }; |
| 15 | |
| 16 | static std::string format_shape(const std::vector<int> & shape) { |
| 17 | std::string out = "["; |
| 18 | for (size_t i = 0; i < shape.size(); ++i) { |
| 19 | if (i > 0) { |
| 20 | out += ", "; |
| 21 | } |
| 22 | out += std::to_string(shape[i]); |
| 23 | } |
| 24 | out += "]"; |
| 25 | return out; |
| 26 | } |
| 27 | |
| 28 | static bool dump_checkpoints(const sam3_state & state, |
| 29 | const std::vector<std::string> & checkpoints, |