#include <std_transforms_sve.hpp>
|
template<typename TIn > |
void | PrepareA (TOperand *out, const TIn *in, const int stride, const int y0, const int ymax, const int k0, const int kmax, int32_t row_sum_multiplier) |
|
template<typename TIn > |
void | PrepareA_indirect (TOperand *out, const TIn *const *const *ptr, size_t stringlen, size_t rounded_stringlen, const int y0, const int ymax, const int k0, const int kmax, int32_t row_sum_multiplier) |
|
template<typename TIn > |
void | PrepareA_convolution (TOperand *out, const TIn *ptr, size_t stride, const convolver< TIn > &conv, size_t rounded_stringlen, const int y0, const int ymax, const int k0, const int kmax, int32_t row_sum_multiplier) |
|
template<typename TIn > |
void | PrepareB (TOperand *out, const TIn *in, const int stride, const int x0, const int xmax, const int k0, const int kmax) |
|
template<typename TOut > |
void | Merge (TOut *out, const TResult *in, int stride, int y0, int ymax, int x0, int xmax, const TOut *bias, const Activation act, bool append) |
|
template<typename TOperand, typename TResult, unsigned int height, unsigned int width_vectors, unsigned int block = 1, unsigned int mmla = 1, bool integrate_sums = false>
class arm_gemm::StdTransformsSVE< TOperand, TResult, height, width_vectors, block, mmla, integrate_sums >
Definition at line 43 of file std_transforms_sve.hpp.
◆ Merge()
void Merge |
( |
TOut * |
out, |
|
|
const TResult * |
in, |
|
|
int |
stride, |
|
|
int |
y0, |
|
|
int |
ymax, |
|
|
int |
x0, |
|
|
int |
xmax, |
|
|
const TOut * |
bias, |
|
|
const Activation |
act, |
|
|
bool |
append |
|
) |
| |
|
inline |
Definition at line 71 of file std_transforms_sve.hpp.
72 MergeResults<width_vectors / mmla, height,
true>(out, in, stride, y0, ymax, x0, xmax,
bias, act, append);
References bias, and arm_gemm::MergeResults().
◆ PrepareA()
void PrepareA |
( |
TOperand * |
out, |
|
|
const TIn * |
in, |
|
|
const int |
stride, |
|
|
const int |
y0, |
|
|
const int |
ymax, |
|
|
const int |
k0, |
|
|
const int |
kmax, |
|
|
int32_t |
row_sum_multiplier |
|
) |
| |
|
inline |
Definition at line 47 of file std_transforms_sve.hpp.
49 Interleave<height, block, VLType::None>(out, in, stride, y0, ymax, k0, kmax, integrate_sums, row_sum_multiplier);
◆ PrepareA_convolution()
void PrepareA_convolution |
( |
TOperand * |
out, |
|
|
const TIn * |
ptr, |
|
|
size_t |
stride, |
|
|
const convolver< TIn > & |
conv, |
|
|
size_t |
rounded_stringlen, |
|
|
const int |
y0, |
|
|
const int |
ymax, |
|
|
const int |
k0, |
|
|
const int |
kmax, |
|
|
int32_t |
row_sum_multiplier |
|
) |
| |
|
inline |
Definition at line 59 of file std_transforms_sve.hpp.
61 ConvolutionInterleave<height, block, VLType::None>(out, ptr, stride, conv, rounded_stringlen, y0, ymax, k0, kmax, integrate_sums, row_sum_multiplier);
◆ PrepareA_indirect()
void PrepareA_indirect |
( |
TOperand * |
out, |
|
|
const TIn *const *const * |
ptr, |
|
|
size_t |
stringlen, |
|
|
size_t |
rounded_stringlen, |
|
|
const int |
y0, |
|
|
const int |
ymax, |
|
|
const int |
k0, |
|
|
const int |
kmax, |
|
|
int32_t |
row_sum_multiplier |
|
) |
| |
|
inline |
Definition at line 53 of file std_transforms_sve.hpp.
55 IndirectInterleave<height, block, VLType::None>(out, ptr, stringlen, rounded_stringlen, y0, ymax, k0, kmax, integrate_sums, row_sum_multiplier);
◆ PrepareB()
void PrepareB |
( |
TOperand * |
out, |
|
|
const TIn * |
in, |
|
|
const int |
stride, |
|
|
const int |
x0, |
|
|
const int |
xmax, |
|
|
const int |
k0, |
|
|
const int |
kmax |
|
) |
| |
|
inline |
Definition at line 65 of file std_transforms_sve.hpp.
67 Transform<width_vectors, block, true, VLType::SVE>(out, in, stride, x0, xmax, k0, kmax);
The documentation for this class was generated from the following file: