dot.h

Go to the documentation of this file.

#pragma once
#include "tensor/tensor.h"
#include "math/matmul.h"
#include "math/scalar_tensor_product.h"

namespace magmadnn {
namespace math {

template <typename T>
void dot(Tensor<T> *A, Tensor<T> *B, Tensor<T> *out);

template <typename T>
void dot(T alpha, Tensor<T> *A, Tensor<T> *B, T beta, Tensor<T> *out);

template <typename T>
void dot(T alpha, bool trans_A, Tensor<T> *A, bool trans_B, Tensor<T> *B, T beta, Tensor<T> *out);

}
}