dotop.h

Go to the documentation of this file.

#pragma once

#include <cstdio>
#include "compute/operation.h"
#include "compute/matmul/matmulop.h"
#include "compute/product/productop.h"
#include "compute/scalarproduct/scalarproductop.h"

namespace magmadnn {
namespace op {

template <typename T>
Operation<T> *dot(Operation<T> *a, Operation<T> *b, bool copy=true, bool needs_grad=true);

}   // namespace op
}   // namespace magmadnn