Class TfLiteIntArrayView

tensorflow/lite/graph_info.cc:27–43

Provide a range iterable wrapper for TfLiteIntArray* (C lists that TfLite C api uses. Can't use the google array_view, since we can't depend on even absl for embedded device reasons. TODO(aselle): Move this into central utilities.

// absl for embedded device reasons.
// TODO(aselle): Move this into central utilities.
class TfLiteIntArrayView {
 public:
  // Construct a view of a TfLiteIntArray*. Note, `int_array` should be non-null
  // and this view does not take ownership of it.
  explicit TfLiteIntArrayView(const TfLiteIntArray* int_array)
      : int_array_(int_array) {}

  typedef const int* const_iterator;
  const_iterator begin() const { return int_array_->data; }
  const_iterator end() const { return &int_array_->data[int_array_->size]; }

  TfLiteIntArrayView(const TfLiteIntArrayView&) = default;
  TfLiteIntArrayView& operator=(const TfLiteIntArrayView& rhs) = default;

 private:
  const TfLiteIntArray* int_array_;
};

// Helper class that actually performs partitioning by node sub set.
// Outputs to a provided `NodeSubset` structure.