Method dim

tensorflow/compiler/xla/array.h:405–408

Returns the size of the dimension at the given index.

  // Returns the size of the dimension at the given index.
  int64 dim(int64 n) const {
    CHECK((size_t)n < sizes_.size());
    return sizes_[n];
  }

  // Returns a vector containing the dimensions of the array.
  const std::vector<int64>& dimensions() const { return sizes_; }