Function get_tensor_dims

kleidiai-examples/audiogen-et/app/main.cpp:188–191

}

static std::vector<executorch::aten::SizesType> get_tensor_dims(const TensorInfo& tensor_info) {
    std::vector<executorch::aten::SizesType> tensor_dims(tensor_info.sizes().begin(), tensor_info.sizes().end());
    return tensor_dims;
}

static size_t get_num_elems(const std::vector<executorch::aten::SizesType>& tensor_dims) {
    size_t numel = 1;