gemm.h File Reference

#include <cstddef>
#include <type_traits>
#include "clustering/always_assert.h"
#include "clustering/math/defaults.h"
#include "clustering/math/thread.h"
#include "clustering/ndarray.h"

Include dependency graph for gemm.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
namespace	clustering
namespace	clustering::math

Functions
template<class T, Layout LA, Layout LB, class Backend = defaults::Backend>
void	clustering::math::gemm (const NDArray< T, 2, LA > &A, const NDArray< T, 2, LB > &B, NDArray< T, 2 > &C, Pool pool, T alpha=T{1}, T beta=T{0})
	One-shot dense matrix-matrix multiply: C := alpha * A * B + beta * C.