Function numElements

compiler/circle-eval-diff/src/Tensor.cpp:27–33

Return number of elements of the node.

// Return number of elements of the node.
uint32_t numElements(const luci::CircleNode *node)
{
  uint32_t num_elem = 1;
  for (uint32_t i = 0; i < node->rank(); ++i)
    num_elem *= node->dim(i).value();
  return num_elem;
}

} // namespace