BatchLinOp Class Reference#
|
Reference API
|
Public Member Functions | |
| size_type | get_num_batch_items () const noexcept |
| dim< 2 > | get_common_size () const |
| const batch_dim< 2 > & | get_size () const noexcept |
| template<typename ValueType > | |
| void | validate_application_parameters (const MultiVector< ValueType > *b, MultiVector< ValueType > *x) const |
| template<typename ValueType > | |
| void | validate_application_parameters (const MultiVector< ValueType > *alpha, const MultiVector< ValueType > *b, const MultiVector< ValueType > *beta, MultiVector< ValueType > *x) const |
Public Member Functions inherited from gko::EnableAbstractPolymorphicObject< BatchLinOp > | |
| std::unique_ptr< BatchLinOp > | create_default (std::shared_ptr< const Executor > exec) const |
| std::unique_ptr< BatchLinOp > | create_default () const |
| std::unique_ptr< BatchLinOp > | clone (std::shared_ptr< const Executor > exec) const |
| std::unique_ptr< BatchLinOp > | clone () const |
| BatchLinOp * | copy_from (const PolymorphicObject *other) |
| std::enable_if_t< std::is_base_of< PolymorphicObject, std::decay_t< Derived > >::value, BatchLinOp > * | copy_from (std::unique_ptr< Derived > &&other) |
| std::enable_if_t< std::is_base_of< PolymorphicObject, std::decay_t< Derived > >::value, BatchLinOp > * | copy_from (const std::unique_ptr< Derived > &other) |
| BatchLinOp * | copy_from (const std::shared_ptr< const PolymorphicObject > &other) |
| BatchLinOp * | move_from (ptr_param< PolymorphicObject > other) |
| BatchLinOp * | clear () |
Public Member Functions inherited from gko::PolymorphicObject | |
| PolymorphicObject & | operator= (const PolymorphicObject &) |
| std::unique_ptr< PolymorphicObject > | create_default (std::shared_ptr< const Executor > exec) const |
| std::unique_ptr< PolymorphicObject > | create_default () const |
| std::unique_ptr< PolymorphicObject > | clone (std::shared_ptr< const Executor > exec) const |
| std::unique_ptr< PolymorphicObject > | clone () const |
| PolymorphicObject * | copy_from (const PolymorphicObject *other) |
| template<typename Derived , typename Deleter > | |
| std::enable_if_t< std::is_base_of< PolymorphicObject, std::decay_t< Derived > >::value, PolymorphicObject > * | copy_from (std::unique_ptr< Derived, Deleter > &&other) |
| template<typename Derived , typename Deleter > | |
| std::enable_if_t< std::is_base_of< PolymorphicObject, std::decay_t< Derived > >::value, PolymorphicObject > * | copy_from (const std::unique_ptr< Derived, Deleter > &other) |
| PolymorphicObject * | copy_from (const std::shared_ptr< const PolymorphicObject > &other) |
| PolymorphicObject * | move_from (ptr_param< PolymorphicObject > other) |
| PolymorphicObject * | clear () |
| std::shared_ptr< const Executor > | get_executor () const noexcept |
Public Member Functions inherited from gko::log::EnableLogging< PolymorphicObject > | |
| void | add_logger (std::shared_ptr< const Logger > logger) override |
| void | remove_logger (const Logger *logger) override |
| void | remove_logger (ptr_param< const Logger > logger) |
| const std::vector< std::shared_ptr< const Logger > > & | get_loggers () const override |
| void | clear_loggers () override |
Public Member Functions inherited from gko::log::Loggable | |
| void | remove_logger (ptr_param< const Logger > logger) |
Protected Member Functions | |
| void | set_size (const batch_dim< 2 > &size) |
| BatchLinOp (std::shared_ptr< const Executor > exec, const batch_dim< 2 > &batch_size) | |
| BatchLinOp (std::shared_ptr< const Executor > exec, const size_type num_batch_items=0, const dim< 2 > &common_size=dim< 2 >{}) | |
Protected Member Functions inherited from gko::PolymorphicObject | |
| PolymorphicObject (std::shared_ptr< const Executor > exec) | |
| PolymorphicObject (const PolymorphicObject &other) | |
| virtual std::unique_ptr< PolymorphicObject > | create_default_impl (std::shared_ptr< const Executor > exec) const =0 |
| virtual PolymorphicObject * | copy_from_impl (const PolymorphicObject *other)=0 |
| virtual PolymorphicObject * | copy_from_impl (std::unique_ptr< PolymorphicObject > other)=0 |
| virtual PolymorphicObject * | move_from_impl (PolymorphicObject *other)=0 |
| virtual PolymorphicObject * | move_from_impl (std::unique_ptr< PolymorphicObject > other)=0 |
| virtual PolymorphicObject * | clear_impl ()=0 |
Protected Member Functions inherited from gko::log::EnableLogging< PolymorphicObject > | |
| void | log (Params &&... params) const |
Additional Inherited Members | |
Protected Attributes inherited from gko::log::EnableLogging< PolymorphicObject > | |
| std::vector< std::shared_ptr< const Logger > > | loggers_ |
Constructor & Destructor Documentation
◆ BatchLinOp() [1/2]
|
inlineexplicitprotected |
Creates a batch operator storing items of uniform sizes.
- Parameters
-
exec the executor where all the operations are performed batch_size the size the batched operator, as a batch_dim object
◆ BatchLinOp() [2/2]
|
inlineexplicitprotected |
Creates a batch operator storing items of uniform sizes.
- Parameters
-
exec the executor where all the operations are performed num_batch_items the number of batch items to be stored in the operator size the common size of the items in the batched operator
Member Function Documentation
◆ get_common_size()
|
inline |
Returns the common size of the batch items.
- Returns
- the common size stored
References gko::batch_dim< Dimensionality, DimensionType >::get_common_size(), and get_size().
Referenced by validate_application_parameters(), and validate_application_parameters().
◆ get_num_batch_items()
|
inlinenoexcept |
Returns the number of items in the batch operator.
- Returns
- number of items in the batch operator
References gko::batch_dim< Dimensionality, DimensionType >::get_num_batch_items(), and get_size().
Referenced by validate_application_parameters(), and validate_application_parameters().
◆ get_size()
|
inlinenoexcept |
Returns the size of the batch operator.
- Returns
- size of the batch operator, a batch_dim object
Referenced by get_common_size(), and get_num_batch_items().
◆ set_size()
|
inlineprotected |
Sets the size of the batch operator.
- Parameters
-
size to be set
◆ validate_application_parameters() [1/2]
|
inline |
Validates the sizes for the apply(alpha, b , beta, x) operation in the concrete BatchLinOp.
References get_common_size(), gko::batch::MultiVector< ValueType >::get_common_size(), gko::batch::MultiVector< ValueType >::get_num_batch_items(), and get_num_batch_items().
◆ validate_application_parameters() [2/2]
|
inline |
Validates the sizes for the apply(b,x) operation in the concrete BatchLinOp.
References get_common_size(), gko::batch::MultiVector< ValueType >::get_common_size(), gko::batch::MultiVector< ValueType >::get_num_batch_items(), and get_num_batch_items().
The documentation for this class was generated from the following file:
- ginkgo/core/base/batch_lin_op.hpp
Generated by
Public Member Functions inherited from