#include <OpMult.hpp>
Inheritance diagram for kp::OpMult:
Collaboration diagram for kp::OpMult:
Public Member Functions | |
| OpMult (std::vector< std::shared_ptr< Tensor >> tensors, std::shared_ptr< Algorithm > algorithm) | |
| ~OpMult () override | |
 Public Member Functions inherited from kp::OpAlgoDispatch | |
| template<typename T = float> | |
| OpAlgoDispatch (const std::shared_ptr< kp::Algorithm > &algorithm, const std::vector< T > &pushConstants={}) | |
| virtual | ~OpAlgoDispatch () override |
| virtual void | record (const vk::CommandBuffer &commandBuffer) override |
| virtual void | preEval (const vk::CommandBuffer &commandBuffer) override |
| virtual void | postEval (const vk::CommandBuffer &commandBuffer) override |
| Public Member Functions inherited from kp::OpBase | |
| virtual | ~OpBase () |
Detailed Description
Operation that performs multiplication on two tensors and outpus on third tensor.
Constructor & Destructor Documentation
◆ OpMult()
|
inline |
Default constructor with parameters that provides the bare minimum requirements for the operations to be able to create and manage their sub-components.
- Parameters
-
tensors Tensors that are to be used in this operation algorithm An algorithm that will be overridden with the OpMult shader data and the tensors provided which are expected to be 3
◆ ~OpMult()
|
inlineoverride |
Default destructor, which is in charge of destroying the algorithm components but does not destroy the underlying tensors
The documentation for this class was generated from the following file:
- /home/alejandro/Programming/kubernetes/seldon/vulkan-kompute/src/include/kompute/operations/OpMult.hpp
