Method dimension

tensorflow/compiler/xla/layout.h:53–53

Returns the bound of the tile in the given dimension index.

  // Returns the bound of the tile in the given dimension index.
  int64 dimension(int i) const { return dimensions_.at(i); }

  // Returns the dimensions of the tile.
  absl::Span<const int64> dimensions() const { return dimensions_; }