33#ifndef GKO_PUBLIC_CORE_BASE_BATCH_DIM_HPP_
34#define GKO_PUBLIC_CORE_BASE_BATCH_DIM_HPP_
40#include <ginkgo/core/base/dim.hpp>
41#include <ginkgo/core/base/types.hpp>
55template <
size_type Dimensionality = 2,
typename DimensionType =
size_type>
87 return x.num_batch_items_ == y.num_batch_items_ &&
88 x.common_size_ == y.common_size_;
114 : common_size_(
dim<dimensionality, dimension_type>{}),
134 dim<dimensionality, dimension_type> common_size_{};
147template <
typename DimensionType>
The Ginkgo namespace.
Definition abstract_factory.hpp:48
constexpr T one()
Returns the multiplicative identity for T.
Definition math.hpp:803
std::size_t size_type
Integral type used for allocation quantities.
Definition types.hpp:120
batch_dim< 2, DimensionType > transpose(const batch_dim< 2, DimensionType > &input)
Returns a batch_dim object with its dimensions swapped for batched operators.
Definition batch_dim.hpp:148
A type representing the dimensions of a multidimensional batch object.
Definition batch_dim.hpp:56
batch_dim(const size_type num_batch_items, const dim< dimensionality, dimension_type > &common_size)
Creates a batch_dim object which stores a uniform size for all batch entries.
Definition batch_dim.hpp:127
friend bool operator!=(const batch_dim< Dimensionality, DimensionType > &x, const batch_dim< Dimensionality, DimensionType > &y)
Checks if two batch_dim objects are different.
Definition batch_dim.hpp:103
dim< dimensionality, dimension_type > get_common_size() const
Get the common size of the batch items.
Definition batch_dim.hpp:72
size_type get_num_batch_items() const
Get the number of batch items stored.
Definition batch_dim.hpp:65
friend bool operator==(const batch_dim &x, const batch_dim &y)
Checks if two batch_dim objects are equal.
Definition batch_dim.hpp:85
batch_dim()
The default constructor.
Definition batch_dim.hpp:113
A type representing the dimensions of a multidimensional object.
Definition dim.hpp:55