dotop.h File Reference

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

Include dependency graph for dotop.h:

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

Go to the source code of this file.

Functions
template<typename T >
Operation< T > *	magmadnn::op::dot (Operation< T > *a, Operation< T > *b, bool copy, bool needs_grad)

Detailed Description

Author

Daniel Nichols

Version

0.1

Date

2019-05-22

Copyright

Copyright (c) 2019

Function Documentation

dot()

template<typename T >

Operation< T > * magmadnn::op::dot	(	Operation< T > *	a,
		Operation< T > *	b,
		bool	copy = true,
		bool	needs_grad = true
	)

Dot operation.

Template Parameters

T	int float double

Parameters

a	tensor
b	tensor
copy
needs_grad

Returns

Operation<T>*