Range.h

Go to the documentation of this file.

 
#pragma once
 
#include <tvm/api.h>
 
#include <Eigen/Core>
 
#include <memory>
#include <string>
#include <vector>
 
namespace tvm
{
class TVM_DLLAPI Range
{
public:
  Range() : start(0), dim(0) {}
  Range(int s, int d) : start(s), dim(d) { assert(d >= 0); }
  int start;
  int dim;
 
  int end() const { return start + dim; }
 
  bool operator==(const Range & other) const { return this->dim == other.dim && this->start == other.start; }
 
  bool operator!=(const Range & other) const { return !operator==(other); }
 
  bool contains(int i) const { return start <= i && i < end(); }
 
  bool contains(const Range & other) const { return this->contains(other.start) && this->end() >= other.end(); }
 
  bool intersects(const Range & other) const { return this->contains(other.start) || other.contains(this->start); }
 
  Range operator+(int i) const { return {this->start + i, this->dim}; }
 
  Range operator-(int i) const { return {this->start - i, this->dim}; }
 
  Range relativeRange(const Range & other) const
  {
    assert(this->contains(other));
    return other - this->start;
  }
};
 
} // namespace tvm