Uni20
Asynchronous tensor and runtime library.
Loading...
Searching...
No Matches
Namespaces | Functions
gemv.hpp File Reference

Fixed-output Tensor GEMV front end. More...

#include <uni20/linalg/backends/cpu/gemv.hpp>
#include <uni20/linalg/dispatch.hpp>
#include <uni20/linalg/operation_tags.hpp>
#include <uni20/tensor/concepts.hpp>
#include <utility>
Include dependency graph for ops/gemv.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

namespace  uni20
 
namespace  uni20::linalg
 

Functions

template<class BackendSelector , uni20::MutableRankedTensorView< 1 > OutputTensor, class Scalar , uni20::RankedTensorView< 2 > MatrixTensor, uni20::RankedTensorView< 1 > InputTensor>
void uni20::linalg::gemv (BackendSelector &&selector, OutputTensor &&output, Scalar alpha, MatrixTensor const &matrix, InputTensor const &input, Scalar beta)
 Update a fixed-size vector as output = alpha * matrix * input + beta * output.
 
template<uni20::MutableRankedTensorView< 1 > OutputTensor, class Scalar , uni20::RankedTensorView< 2 > MatrixTensor, uni20::RankedTensorView< 1 > InputTensor>
void uni20::linalg::gemv (OutputTensor &&output, Scalar alpha, MatrixTensor const &matrix, InputTensor const &input, Scalar beta)
 Apply the fixed-storage gemv contract using the operands' default backend selector.
 

Detailed Description

Fixed-output Tensor GEMV front end.