diff options
Diffstat (limited to 'reference_model/src/generate/generate_utils.cc')
| -rw-r--r-- | reference_model/src/generate/generate_utils.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/reference_model/src/generate/generate_utils.cc b/reference_model/src/generate/generate_utils.cc index c52f051..c32d0fb 100644 --- a/reference_model/src/generate/generate_utils.cc +++ b/reference_model/src/generate/generate_utils.cc @@ -110,6 +110,8 @@ std::optional<GenerateConfig> parseGenerateConfig(const char* json, const char* int64_t numElementsFromShape(const std::vector<int32_t>& shape) { + // Rank 0 shapes have no entries and so this will return 1 + // Other ranked shapes will return the product of their dimensions return std::accumulate(std::begin(shape), std::end(shape), 1, std::multiplies<int64_t>()); } |