matmulop.cpp File Reference

#include "compute/matmul/matmulop.h"

Include dependency graph for matmulop.cpp:

Functions
template<typename T >
MatmulOp< T > *	magmadnn::op::matmul (Operation< T > *a, Operation< T > *b, bool needs_grad)
template MatmulOp< int > *	magmadnn::op::matmul (Operation< int > *a, Operation< int > *b, bool needs_grad)
template MatmulOp< float > *	magmadnn::op::matmul (Operation< float > *a, Operation< float > *b, bool needs_grad)
template MatmulOp< double > *	magmadnn::op::matmul (Operation< double > *a, Operation< double > *b, bool needs_grad)
template<typename T >
MatmulOp< T > *	magmadnn::op::matmul (T alpha, Operation< T > *a, Operation< T > *b, T beta, Operation< T > *c, bool copy, bool needs_grad)
template MatmulOp< int > *	magmadnn::op::matmul (int alpha, Operation< int > *a, Operation< int > *b, int beta, Operation< int > *c, bool copy, bool needs_grad)
template MatmulOp< float > *	magmadnn::op::matmul (float alpha, Operation< float > *a, Operation< float > *b, float beta, Operation< float > *c, bool copy, bool needs_grad)
template MatmulOp< double > *	magmadnn::op::matmul (double alpha, Operation< double > *a, Operation< double > *b, double beta, Operation< double > *c, bool copy, bool needs_grad)

Detailed Description

Author

Daniel Nichols

Version

1.0

Date

2019-02-20

Copyright

Copyright (c) 2019

Function Documentation

matmul()

template<typename T >

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

Returns a new operation of type matmul. It computes the matrix product of A and B.

Template Parameters

T

Parameters

a
b
copy

Returns

MatmulOp<T>*

Here is the caller graph for this function: