| 191 | } |
| 192 | |
| 193 | static size_t get_num_elems(const std::vector<executorch::aten::SizesType>& tensor_dims) { |
| 194 | size_t numel = 1; |
| 195 | for (const auto& dim : tensor_dims) { |
| 196 | numel *= dim; |
| 197 | } |
| 198 | return numel; |
| 199 | } |
| 200 | |
| 201 | static void dry_run(std::unique_ptr<executorch::extension::Module>& module) { |
| 202 |