|
<html><head><title>dlib C++ Library - cublas_dlibapi.h</title></head><body bgcolor='white'><pre> |
|
<font color='#009900'>// Copyright (C) 2015 Davis E. King ([email protected]) |
|
</font><font color='#009900'>// License: Boost Software License See LICENSE.txt for the full license. |
|
</font><font color='#0000FF'>#ifndef</font> DLIB_DNN_CuBLAS_H_ |
|
<font color='#0000FF'>#define</font> DLIB_DNN_CuBLAS_H_ |
|
|
|
<font color='#0000FF'>#ifdef</font> DLIB_USE_CUDA |
|
|
|
<font color='#0000FF'>#include</font> "<a style='text-decoration:none' href='tensor.h.html'>tensor.h</a>" |
|
<font color='#0000FF'>#include</font> "<a style='text-decoration:none' href='cuda_errors.h.html'>cuda_errors.h</a>" |
|
|
|
<font color='#0000FF'>namespace</font> dlib |
|
<b>{</b> |
|
<font color='#0000FF'>namespace</font> cuda |
|
<b>{</b> |
|
|
|
<font color='#009900'>// ----------------------------------------------------------------------------------- |
|
</font> |
|
<font color='#0000FF'><u>void</u></font> <b><a name='gemm'></a>gemm</b> <font face='Lucida Console'>(</font> |
|
<font color='#0000FF'><u>float</u></font> beta, |
|
tensor<font color='#5555FF'>&</font> dest, |
|
<font color='#0000FF'><u>float</u></font> alpha, |
|
<font color='#0000FF'>const</font> tensor<font color='#5555FF'>&</font> lhs, |
|
<font color='#0000FF'><u>bool</u></font> trans_lhs, |
|
<font color='#0000FF'>const</font> tensor<font color='#5555FF'>&</font> rhs, |
|
<font color='#0000FF'><u>bool</u></font> trans_rhs |
|
<font face='Lucida Console'>)</font>; |
|
<font color='#009900'>/*! |
|
requires |
|
- The dimensions of lhs and rhs must be compatible for matrix |
|
multiplication. In particular: |
|
- Let L == trans_lhs ? trans(mat(lhs)) : mat(lhs) |
|
- Let R == trans_rhs ? trans(mat(rhs)) : mat(rhs) |
|
- Let D == mat(dest) |
|
- D.nr() == L.nr() && D.nc() == R.nc() |
|
(i.e. dest must be preallocated and have the correct output dimensions) |
|
- L.nc() == R.nr() |
|
ensures |
|
- performs: dest = alpha*L*R + beta*mat(dest) |
|
!*/</font> |
|
|
|
<font color='#009900'>// ------------------------------------------------------------------------------------ |
|
</font> |
|
<b>}</b> |
|
<b>}</b> |
|
|
|
<font color='#0000FF'>#endif</font> <font color='#009900'>// DLIB_USE_CUDA |
|
</font> |
|
<font color='#0000FF'>#endif</font> <font color='#009900'>// DLIB_DNN_CuBLAS_H_ |
|
</font> |
|
|
|
|
|
</pre></body></html> |
