| <!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd"> |
| <!-- NewPage --> |
| <html lang="en"> |
| <head> |
| <!-- Generated by javadoc --> |
| <meta http-equiv="Content-Type" content="text/html; charset=UTF-8"> |
| <title>LibMatrixCuDNN (SystemDS 2.1.0-SNAPSHOT API)</title> |
| <link rel="stylesheet" type="text/css" href="../../../../../../stylesheet.css" title="Style"> |
| <script type="text/javascript" src="../../../../../../script.js"></script> |
| </head> |
| <body> |
| <script type="text/javascript"><!-- |
| try { |
| if (location.href.indexOf('is-external=true') == -1) { |
| parent.document.title="LibMatrixCuDNN (SystemDS 2.1.0-SNAPSHOT API)"; |
| } |
| } |
| catch(err) { |
| } |
| //--> |
| var methods = {"i0":9,"i1":9,"i2":9,"i3":9,"i4":9,"i5":9,"i6":9,"i7":9,"i8":9,"i9":9,"i10":9,"i11":9,"i12":9,"i13":9}; |
| var tabs = {65535:["t0","All Methods"],1:["t1","Static Methods"],8:["t4","Concrete Methods"]}; |
| var altColor = "altColor"; |
| var rowColor = "rowColor"; |
| var tableTab = "tableTab"; |
| var activeTableTab = "activeTableTab"; |
| </script> |
| <noscript> |
| <div>JavaScript is disabled on your browser.</div> |
| </noscript> |
| <!-- ========= START OF TOP NAVBAR ======= --> |
| <div class="topNav"><a name="navbar.top"> |
| <!-- --> |
| </a> |
| <div class="skipNav"><a href="#skip.navbar.top" title="Skip navigation links">Skip navigation links</a></div> |
| <a name="navbar.top.firstrow"> |
| <!-- --> |
| </a> |
| <ul class="navList" title="Navigation"> |
| <li><a href="../../../../../../overview-summary.html">Overview</a></li> |
| <li><a href="package-summary.html">Package</a></li> |
| <li class="navBarCell1Rev">Class</li> |
| <li><a href="class-use/LibMatrixCuDNN.html">Use</a></li> |
| <li><a href="package-tree.html">Tree</a></li> |
| <li><a href="../../../../../../deprecated-list.html">Deprecated</a></li> |
| <li><a href="../../../../../../index-all.html">Index</a></li> |
| <li><a href="../../../../../../help-doc.html">Help</a></li> |
| </ul> |
| </div> |
| <div class="subNav"> |
| <ul class="navList"> |
| <li><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html" title="class in org.apache.sysds.runtime.matrix.data"><span class="typeNameLink">Prev Class</span></a></li> |
| <li><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNNConvolutionAlgorithm.html" title="class in org.apache.sysds.runtime.matrix.data"><span class="typeNameLink">Next Class</span></a></li> |
| </ul> |
| <ul class="navList"> |
| <li><a href="../../../../../../index.html?org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html" target="_top">Frames</a></li> |
| <li><a href="LibMatrixCuDNN.html" target="_top">No Frames</a></li> |
| </ul> |
| <ul class="navList" id="allclasses_navbar_top"> |
| <li><a href="../../../../../../allclasses-noframe.html">All Classes</a></li> |
| </ul> |
| <div> |
| <script type="text/javascript"><!-- |
| allClassesLink = document.getElementById("allclasses_navbar_top"); |
| if(window==top) { |
| allClassesLink.style.display = "block"; |
| } |
| else { |
| allClassesLink.style.display = "none"; |
| } |
| //--> |
| </script> |
| </div> |
| <div> |
| <ul class="subNavList"> |
| <li>Summary: </li> |
| <li>Nested | </li> |
| <li><a href="#fields.inherited.from.class.org.apache.sysds.runtime.matrix.data.LibMatrixCUDA">Field</a> | </li> |
| <li><a href="#constructor.summary">Constr</a> | </li> |
| <li><a href="#method.summary">Method</a></li> |
| </ul> |
| <ul class="subNavList"> |
| <li>Detail: </li> |
| <li>Field | </li> |
| <li><a href="#constructor.detail">Constr</a> | </li> |
| <li><a href="#method.detail">Method</a></li> |
| </ul> |
| </div> |
| <a name="skip.navbar.top"> |
| <!-- --> |
| </a></div> |
| <!-- ========= END OF TOP NAVBAR ========= --> |
| <!-- ======== START OF CLASS DATA ======== --> |
| <div class="header"> |
| <div class="subTitle">org.apache.sysds.runtime.matrix.data</div> |
| <h2 title="Class LibMatrixCuDNN" class="title">Class LibMatrixCuDNN</h2> |
| </div> |
| <div class="contentContainer"> |
| <ul class="inheritance"> |
| <li><a href="https://docs.oracle.com/javase/8/docs/api/java/lang/Object.html?is-external=true" title="class or interface in java.lang">java.lang.Object</a></li> |
| <li> |
| <ul class="inheritance"> |
| <li><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html" title="class in org.apache.sysds.runtime.matrix.data">org.apache.sysds.runtime.matrix.data.LibMatrixCUDA</a></li> |
| <li> |
| <ul class="inheritance"> |
| <li>org.apache.sysds.runtime.matrix.data.LibMatrixCuDNN</li> |
| </ul> |
| </li> |
| </ul> |
| </li> |
| </ul> |
| <div class="description"> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <hr> |
| <br> |
| <pre>public class <span class="typeNameLabel">LibMatrixCuDNN</span> |
| extends <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html" title="class in org.apache.sysds.runtime.matrix.data">LibMatrixCUDA</a></pre> |
| <div class="block">This class contains method that invoke CuDNN operations.</div> |
| </li> |
| </ul> |
| </div> |
| <div class="summary"> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <!-- =========== FIELD SUMMARY =========== --> |
| <ul class="blockList"> |
| <li class="blockList"><a name="field.summary"> |
| <!-- --> |
| </a> |
| <h3>Field Summary</h3> |
| <ul class="blockList"> |
| <li class="blockList"><a name="fields.inherited.from.class.org.apache.sysds.runtime.matrix.data.LibMatrixCUDA"> |
| <!-- --> |
| </a> |
| <h3>Fields inherited from class org.apache.sysds.runtime.matrix.data.<a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html" title="class in org.apache.sysds.runtime.matrix.data">LibMatrixCUDA</a></h3> |
| <code><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#cudaSupportFunctions">cudaSupportFunctions</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#customKernelSuffix">customKernelSuffix</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#sizeOfDataType">sizeOfDataType</a></code></li> |
| </ul> |
| </li> |
| </ul> |
| <!-- ======== CONSTRUCTOR SUMMARY ======== --> |
| <ul class="blockList"> |
| <li class="blockList"><a name="constructor.summary"> |
| <!-- --> |
| </a> |
| <h3>Constructor Summary</h3> |
| <table class="memberSummary" border="0" cellpadding="3" cellspacing="0" summary="Constructor Summary table, listing constructors, and an explanation"> |
| <caption><span>Constructors</span><span class="tabEnd"> </span></caption> |
| <tr> |
| <th class="colOne" scope="col">Constructor and Description</th> |
| </tr> |
| <tr class="altColor"> |
| <td class="colOne"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#LibMatrixCuDNN--">LibMatrixCuDNN</a></span>()</code> </td> |
| </tr> |
| </table> |
| </li> |
| </ul> |
| <!-- ========== METHOD SUMMARY =========== --> |
| <ul class="blockList"> |
| <li class="blockList"><a name="method.summary"> |
| <!-- --> |
| </a> |
| <h3>Method Summary</h3> |
| <table class="memberSummary" border="0" cellpadding="3" cellspacing="0" summary="Method Summary table, listing methods, and an explanation"> |
| <caption><span id="t0" class="activeTableTab"><span>All Methods</span><span class="tabEnd"> </span></span><span id="t1" class="tableTab"><span><a href="javascript:show(1);">Static Methods</a></span><span class="tabEnd"> </span></span><span id="t4" class="tableTab"><span><a href="javascript:show(8);">Concrete Methods</a></span><span class="tabEnd"> </span></span></caption> |
| <tr> |
| <th class="colFirst" scope="col">Modifier and Type</th> |
| <th class="colLast" scope="col">Method and Description</th> |
| </tr> |
| <tr id="i0" class="altColor"> |
| <td class="colFirst"><code>static void</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#batchNormalizationBackward-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-double-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-">batchNormalizationBackward</a></span>(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dout, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> scale, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dX, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dScale, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dBias, |
| double epsilon, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> resultSaveMean, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> resultSaveInvVariance)</code> |
| <div class="block">This method computes the backpropagation errors for image, scale and bias of batch normalization layer</div> |
| </td> |
| </tr> |
| <tr id="i1" class="rowColor"> |
| <td class="colFirst"><code>static void</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#batchNormalizationForwardInference-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-double-">batchNormalizationForwardInference</a></span>(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> scale, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> bias, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> runningMean, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> runningVar, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> ret, |
| double epsilon)</code> |
| <div class="block">Performs the forward BatchNormalization layer computation for inference</div> |
| </td> |
| </tr> |
| <tr id="i2" class="altColor"> |
| <td class="colFirst"><code>static void</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#batchNormalizationForwardTraining-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-double-double-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-">batchNormalizationForwardTraining</a></span>(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> scale, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> bias, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> runningMean, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> runningVar, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> ret, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> retRunningMean, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> retRunningVar, |
| double epsilon, |
| double exponentialAverageFactor, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> resultSaveMean, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> resultSaveInvVariance)</code> |
| <div class="block">Performs the forward BatchNormalization layer computation for training</div> |
| </td> |
| </tr> |
| <tr id="i3" class="rowColor"> |
| <td class="colFirst"><code>static void</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#conv2d-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-int-int-int-int-int-int-int-int-int-int-int-int-int-double-">conv2d</a></span>(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> filter, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> outputBlock, |
| int N, |
| int C, |
| int H, |
| int W, |
| int K, |
| int R, |
| int S, |
| int pad_h, |
| int pad_w, |
| int stride_h, |
| int stride_w, |
| int P, |
| int Q, |
| double intermediateMemoryBudget)</code> |
| <div class="block">Performs a 2D convolution</div> |
| </td> |
| </tr> |
| <tr id="i4" class="altColor"> |
| <td class="colFirst"><code>static void</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#conv2dBackwardData-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-int-int-int-int-int-int-int-int-int-int-int-int-int-double-">conv2dBackwardData</a></span>(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> filter, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dout, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> output, |
| int N, |
| int C, |
| int H, |
| int W, |
| int K, |
| int R, |
| int S, |
| int pad_h, |
| int pad_w, |
| int stride_h, |
| int stride_w, |
| int P, |
| int Q, |
| double intermediateMemoryBudget)</code> |
| <div class="block">This method computes the backpropogation errors for previous layer of convolution operation</div> |
| </td> |
| </tr> |
| <tr id="i5" class="rowColor"> |
| <td class="colFirst"><code>static void</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#conv2dBackwardFilter-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-int-int-int-int-int-int-int-int-int-int-int-int-int-double-">conv2dBackwardFilter</a></span>(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dout, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> outputBlock, |
| int N, |
| int C, |
| int H, |
| int W, |
| int K, |
| int R, |
| int S, |
| int pad_h, |
| int pad_w, |
| int stride_h, |
| int stride_w, |
| int P, |
| int Q, |
| double intermediateMemoryBudget)</code> |
| <div class="block">This method computes the backpropogation errors for filter of convolution operation</div> |
| </td> |
| </tr> |
| <tr id="i6" class="altColor"> |
| <td class="colFirst"><code>static void</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#conv2dBiasAdd-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-int-int-int-int-int-int-int-int-int-int-int-int-int-double-">conv2dBiasAdd</a></span>(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> bias, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> filter, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> output, |
| int N, |
| int C, |
| int H, |
| int W, |
| int K, |
| int R, |
| int S, |
| int pad_h, |
| int pad_w, |
| int stride_h, |
| int stride_w, |
| int P, |
| int Q, |
| double intermediateMemoryBudget)</code> |
| <div class="block">Does a 2D convolution followed by a bias_add</div> |
| </td> |
| </tr> |
| <tr id="i7" class="rowColor"> |
| <td class="colFirst"><code>static jcuda.Pointer</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#getDensePointerForCuDNN-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-int-int-">getDensePointerForCuDNN</a></span>(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| int numRows, |
| int numCols)</code> |
| <div class="block">Convenience method to get jcudaDenseMatrixPtr.</div> |
| </td> |
| </tr> |
| <tr id="i8" class="altColor"> |
| <td class="colFirst"><code>static void</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#lstm-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-jcuda.Pointer-jcuda.Pointer-jcuda.Pointer-jcuda.Pointer-boolean-java.lang.String-java.lang.String-int-int-int-int-">lstm</a></span>(<a href="../../../../../../org/apache/sysds/runtime/controlprogram/context/ExecutionContext.html" title="class in org.apache.sysds.runtime.controlprogram.context">ExecutionContext</a> ec, |
| <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| jcuda.Pointer X, |
| jcuda.Pointer wPointer, |
| jcuda.Pointer out0, |
| jcuda.Pointer c0, |
| boolean return_sequences, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> outputName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> cyName, |
| int N, |
| int M, |
| int D, |
| int T)</code> |
| <div class="block">Computes the forward pass for an LSTM layer with M neurons.</div> |
| </td> |
| </tr> |
| <tr id="i9" class="rowColor"> |
| <td class="colFirst"><code>static void</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#lstmBackward-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-jcuda.Pointer-jcuda.Pointer-jcuda.Pointer-jcuda.Pointer-java.lang.String-java.lang.String-java.lang.String-java.lang.String-java.lang.String-java.lang.String-java.lang.String-boolean-int-int-int-int-">lstmBackward</a></span>(<a href="../../../../../../org/apache/sysds/runtime/controlprogram/context/ExecutionContext.html" title="class in org.apache.sysds.runtime.controlprogram.context">ExecutionContext</a> ec, |
| <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| jcuda.Pointer x, |
| jcuda.Pointer hx, |
| jcuda.Pointer cx, |
| jcuda.Pointer wPointer, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> doutName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> dcyName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> dxName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> dwName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> dbName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> dhxName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> dcxName, |
| boolean return_sequences, |
| int N, |
| int M, |
| int D, |
| int T)</code> </td> |
| </tr> |
| <tr id="i10" class="altColor"> |
| <td class="colFirst"><code>static void</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#pooling-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-int-int-int-int-int-int-int-int-int-int-int-int-int-org.apache.sysds.runtime.matrix.data.LibMatrixDNN.PoolingType-double-">pooling</a></span>(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> outputBlock, |
| int N, |
| int C, |
| int H, |
| int W, |
| int K, |
| int R, |
| int S, |
| int pad_h, |
| int pad_w, |
| int stride_h, |
| int stride_w, |
| int P, |
| int Q, |
| <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixDNN.PoolingType.html" title="enum in org.apache.sysds.runtime.matrix.data">LibMatrixDNN.PoolingType</a> poolingType, |
| double intermediateMemoryBudget)</code> |
| <div class="block">performs maxpooling on GPU by exploiting cudnnPoolingForward(...)</div> |
| </td> |
| </tr> |
| <tr id="i11" class="rowColor"> |
| <td class="colFirst"><code>static void</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#poolingBackward-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-int-int-int-int-int-int-int-int-int-int-int-int-int-org.apache.sysds.runtime.matrix.data.LibMatrixDNN.PoolingType-double-">poolingBackward</a></span>(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dout, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> maxpoolOutput, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> outputBlock, |
| int N, |
| int C, |
| int H, |
| int W, |
| int K, |
| int R, |
| int S, |
| int pad_h, |
| int pad_w, |
| int stride_h, |
| int stride_w, |
| int P, |
| int Q, |
| <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixDNN.PoolingType.html" title="enum in org.apache.sysds.runtime.matrix.data">LibMatrixDNN.PoolingType</a> poolingType, |
| double intermediateMemoryBudget)</code> |
| <div class="block">Performs maxpoolingBackward on GPU by exploiting cudnnPoolingBackward(...) |
| This method computes the backpropogation errors for previous layer of maxpooling operation</div> |
| </td> |
| </tr> |
| <tr id="i12" class="altColor"> |
| <td class="colFirst"><code>static void</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#relu-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">relu</a></span>(<a href="../../../../../../org/apache/sysds/runtime/controlprogram/context/ExecutionContext.html" title="class in org.apache.sysds.runtime.controlprogram.context">ExecutionContext</a> ec, |
| <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> in, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> outputName)</code> |
| <div class="block">Performs the relu operation on the GPU.</div> |
| </td> |
| </tr> |
| <tr id="i13" class="rowColor"> |
| <td class="colFirst"><code>static void</code></td> |
| <td class="colLast"><code><span class="memberNameLink"><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html#softmax-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">softmax</a></span>(<a href="../../../../../../org/apache/sysds/runtime/controlprogram/context/ExecutionContext.html" title="class in org.apache.sysds.runtime.controlprogram.context">ExecutionContext</a> ec, |
| <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> in1, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> outputName)</code> |
| <div class="block">Performs an "softmax" operation on a matrix on the GPU</div> |
| </td> |
| </tr> |
| </table> |
| <ul class="blockList"> |
| <li class="blockList"><a name="methods.inherited.from.class.org.apache.sysds.runtime.matrix.data.LibMatrixCUDA"> |
| <!-- --> |
| </a> |
| <h3>Methods inherited from class org.apache.sysds.runtime.matrix.data.<a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html" title="class in org.apache.sysds.runtime.matrix.data">LibMatrixCUDA</a></h3> |
| <code><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#abs-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">abs</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#acos-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">acos</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#asin-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">asin</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#atan-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">atan</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#axpy-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-double-">axpy</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#biasAdd-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-">biasAdd</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#biasMultiply-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-">biasMultiply</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#cbind-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">cbind</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#ceil-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">ceil</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#channelSums-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-long-long-">channelSums</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#computeNNZ-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-jcuda.Pointer-int-">computeNNZ</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#cos-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">cos</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#cosh-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">cosh</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#cumulativeScan-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">cumulativeScan</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#cumulativeSumProduct-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">cumulativeSumProduct</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#denseTranspose-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-jcuda.Pointer-jcuda.Pointer-long-long-">denseTranspose</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#deviceCopy-java.lang.String-jcuda.Pointer-jcuda.Pointer-int-int-">deviceCopy</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#double2float-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-jcuda.Pointer-jcuda.Pointer-int-">double2float</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#exp-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">exp</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#float2double-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-jcuda.Pointer-jcuda.Pointer-int-">float2double</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#floor-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">floor</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#getCudaKernels-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-">getCudaKernels</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#getDenseMatrixOutputForGPUInstruction-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-java.lang.String-java.lang.String-long-long-">getDenseMatrixOutputForGPUInstruction</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#getDensePointer-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">getDensePointer</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#getNnz-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-boolean-">getNnz</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#isInSparseFormat-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-">isInSparseFormat</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#log-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">log</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#matmultTSMM-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-boolean-">matmultTSMM</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#matrixMatrixArithmetic-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-boolean-boolean-org.apache.sysds.runtime.matrix.operators.BinaryOperator-">matrixMatrixArithmetic</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#matrixMatrixRelational-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-org.apache.sysds.runtime.matrix.operators.BinaryOperator-">matrixMatrixRelational</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#matrixScalarArithmetic-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-boolean-org.apache.sysds.runtime.matrix.operators.ScalarOperator-">matrixScalarArithmetic</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#matrixScalarOp-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-boolean-org.apache.sysds.runtime.matrix.operators.ScalarOperator-">matrixScalarOp</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#matrixScalarRelational-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-org.apache.sysds.runtime.matrix.operators.ScalarOperator-">matrixScalarRelational</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#one--">one</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#rbind-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">rbind</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#reluBackward-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-">reluBackward</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#resetFloatingPointPrecision--">resetFloatingPointPrecision</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#round-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">round</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#sigmoid-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">sigmoid</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#sign-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">sign</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#sin-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">sin</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#sinh-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">sinh</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#sliceOperations-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.util.IndexRange-java.lang.String-">sliceOperations</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#solve-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">solve</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#sqrt-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">sqrt</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#tan-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">tan</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#tanh-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">tanh</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#toInt-long-">toInt</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#transpose-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-">transpose</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#unaryAggregate-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-org.apache.sysds.runtime.matrix.operators.AggregateUnaryOperator-">unaryAggregate</a>, <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html#zero--">zero</a></code></li> |
| </ul> |
| <ul class="blockList"> |
| <li class="blockList"><a name="methods.inherited.from.class.java.lang.Object"> |
| <!-- --> |
| </a> |
| <h3>Methods inherited from class java.lang.<a href="https://docs.oracle.com/javase/8/docs/api/java/lang/Object.html?is-external=true" title="class or interface in java.lang">Object</a></h3> |
| <code><a href="https://docs.oracle.com/javase/8/docs/api/java/lang/Object.html?is-external=true#equals-java.lang.Object-" title="class or interface in java.lang">equals</a>, <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/Object.html?is-external=true#getClass--" title="class or interface in java.lang">getClass</a>, <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/Object.html?is-external=true#hashCode--" title="class or interface in java.lang">hashCode</a>, <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/Object.html?is-external=true#notify--" title="class or interface in java.lang">notify</a>, <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/Object.html?is-external=true#notifyAll--" title="class or interface in java.lang">notifyAll</a>, <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/Object.html?is-external=true#toString--" title="class or interface in java.lang">toString</a>, <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/Object.html?is-external=true#wait--" title="class or interface in java.lang">wait</a>, <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/Object.html?is-external=true#wait-long-" title="class or interface in java.lang">wait</a>, <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/Object.html?is-external=true#wait-long-int-" title="class or interface in java.lang">wait</a></code></li> |
| </ul> |
| </li> |
| </ul> |
| </li> |
| </ul> |
| </div> |
| <div class="details"> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <!-- ========= CONSTRUCTOR DETAIL ======== --> |
| <ul class="blockList"> |
| <li class="blockList"><a name="constructor.detail"> |
| <!-- --> |
| </a> |
| <h3>Constructor Detail</h3> |
| <a name="LibMatrixCuDNN--"> |
| <!-- --> |
| </a> |
| <ul class="blockListLast"> |
| <li class="blockList"> |
| <h4>LibMatrixCuDNN</h4> |
| <pre>public LibMatrixCuDNN()</pre> |
| </li> |
| </ul> |
| </li> |
| </ul> |
| <!-- ============ METHOD DETAIL ========== --> |
| <ul class="blockList"> |
| <li class="blockList"><a name="method.detail"> |
| <!-- --> |
| </a> |
| <h3>Method Detail</h3> |
| <a name="conv2dBiasAdd-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-int-int-int-int-int-int-int-int-int-int-int-int-int-double-"> |
| <!-- --> |
| </a> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <h4>conv2dBiasAdd</h4> |
| <pre>public static void conv2dBiasAdd(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> bias, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> filter, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> output, |
| int N, |
| int C, |
| int H, |
| int W, |
| int K, |
| int R, |
| int S, |
| int pad_h, |
| int pad_w, |
| int stride_h, |
| int stride_w, |
| int P, |
| int Q, |
| double intermediateMemoryBudget)</pre> |
| <div class="block">Does a 2D convolution followed by a bias_add</div> |
| <dl> |
| <dt><span class="paramLabel">Parameters:</span></dt> |
| <dd><code>gCtx</code> - a valid <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context"><code>GPUContext</code></a></dd> |
| <dd><code>instName</code> - the invoking instruction's name for record <a href="../../../../../../org/apache/sysds/utils/Statistics.html" title="class in org.apache.sysds.utils"><code>Statistics</code></a>.</dd> |
| <dd><code>image</code> - input image matrix object</dd> |
| <dd><code>bias</code> - bias matrix object</dd> |
| <dd><code>filter</code> - filter matrix object</dd> |
| <dd><code>output</code> - output matrix object</dd> |
| <dd><code>N</code> - number of input images</dd> |
| <dd><code>C</code> - number of channels</dd> |
| <dd><code>H</code> - height of each image</dd> |
| <dd><code>W</code> - width of each image</dd> |
| <dd><code>K</code> - number of output "channels"</dd> |
| <dd><code>R</code> - height of filter</dd> |
| <dd><code>S</code> - width of filter</dd> |
| <dd><code>pad_h</code> - padding height</dd> |
| <dd><code>pad_w</code> - padding width</dd> |
| <dd><code>stride_h</code> - stride height</dd> |
| <dd><code>stride_w</code> - string width</dd> |
| <dd><code>P</code> - output height</dd> |
| <dd><code>Q</code> - output width</dd> |
| <dd><code>intermediateMemoryBudget</code> - intermediate memory budget</dd> |
| </dl> |
| </li> |
| </ul> |
| <a name="conv2d-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-int-int-int-int-int-int-int-int-int-int-int-int-int-double-"> |
| <!-- --> |
| </a> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <h4>conv2d</h4> |
| <pre>public static void conv2d(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> filter, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> outputBlock, |
| int N, |
| int C, |
| int H, |
| int W, |
| int K, |
| int R, |
| int S, |
| int pad_h, |
| int pad_w, |
| int stride_h, |
| int stride_w, |
| int P, |
| int Q, |
| double intermediateMemoryBudget)</pre> |
| <div class="block">Performs a 2D convolution</div> |
| <dl> |
| <dt><span class="paramLabel">Parameters:</span></dt> |
| <dd><code>gCtx</code> - a valid <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context"><code>GPUContext</code></a></dd> |
| <dd><code>instName</code> - the invoking instruction's name for record <a href="../../../../../../org/apache/sysds/utils/Statistics.html" title="class in org.apache.sysds.utils"><code>Statistics</code></a>.</dd> |
| <dd><code>image</code> - input matrix object</dd> |
| <dd><code>filter</code> - filter matrix object</dd> |
| <dd><code>outputBlock</code> - output matrix object</dd> |
| <dd><code>N</code> - number of input images</dd> |
| <dd><code>C</code> - number of channels</dd> |
| <dd><code>H</code> - height of each image</dd> |
| <dd><code>W</code> - width of each image</dd> |
| <dd><code>K</code> - number of output "channels"</dd> |
| <dd><code>R</code> - height of filter</dd> |
| <dd><code>S</code> - width of filter</dd> |
| <dd><code>pad_h</code> - padding height</dd> |
| <dd><code>pad_w</code> - padding width</dd> |
| <dd><code>stride_h</code> - stride height</dd> |
| <dd><code>stride_w</code> - string width</dd> |
| <dd><code>P</code> - output height</dd> |
| <dd><code>Q</code> - output width</dd> |
| <dd><code>intermediateMemoryBudget</code> - intermediate memory budget</dd> |
| </dl> |
| </li> |
| </ul> |
| <a name="softmax-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-"> |
| <!-- --> |
| </a> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <h4>softmax</h4> |
| <pre>public static void softmax(<a href="../../../../../../org/apache/sysds/runtime/controlprogram/context/ExecutionContext.html" title="class in org.apache.sysds.runtime.controlprogram.context">ExecutionContext</a> ec, |
| <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> in1, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> outputName)</pre> |
| <div class="block">Performs an "softmax" operation on a matrix on the GPU</div> |
| <dl> |
| <dt><span class="paramLabel">Parameters:</span></dt> |
| <dd><code>ec</code> - execution context</dd> |
| <dd><code>gCtx</code> - a valid <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context"><code>GPUContext</code></a></dd> |
| <dd><code>instName</code> - the invoking instruction's name for record <a href="../../../../../../org/apache/sysds/utils/Statistics.html" title="class in org.apache.sysds.utils"><code>Statistics</code></a>.</dd> |
| <dd><code>in1</code> - input matrix</dd> |
| <dd><code>outputName</code> - output matrix name</dd> |
| </dl> |
| </li> |
| </ul> |
| <a name="conv2dBackwardFilter-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-int-int-int-int-int-int-int-int-int-int-int-int-int-double-"> |
| <!-- --> |
| </a> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <h4>conv2dBackwardFilter</h4> |
| <pre>public static void conv2dBackwardFilter(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dout, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> outputBlock, |
| int N, |
| int C, |
| int H, |
| int W, |
| int K, |
| int R, |
| int S, |
| int pad_h, |
| int pad_w, |
| int stride_h, |
| int stride_w, |
| int P, |
| int Q, |
| double intermediateMemoryBudget)</pre> |
| <div class="block">This method computes the backpropogation errors for filter of convolution operation</div> |
| <dl> |
| <dt><span class="paramLabel">Parameters:</span></dt> |
| <dd><code>gCtx</code> - a valid <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context"><code>GPUContext</code></a></dd> |
| <dd><code>instName</code> - the invoking instruction's name for record <a href="../../../../../../org/apache/sysds/utils/Statistics.html" title="class in org.apache.sysds.utils"><code>Statistics</code></a>.</dd> |
| <dd><code>image</code> - input image</dd> |
| <dd><code>dout</code> - errors from next layer</dd> |
| <dd><code>outputBlock</code> - output errors</dd> |
| <dd><code>N</code> - number of images</dd> |
| <dd><code>C</code> - number of channels</dd> |
| <dd><code>H</code> - height</dd> |
| <dd><code>W</code> - width</dd> |
| <dd><code>K</code> - number of filters</dd> |
| <dd><code>R</code> - filter height</dd> |
| <dd><code>S</code> - filter width</dd> |
| <dd><code>pad_h</code> - pad height</dd> |
| <dd><code>pad_w</code> - pad width</dd> |
| <dd><code>stride_h</code> - stride height</dd> |
| <dd><code>stride_w</code> - stride width</dd> |
| <dd><code>P</code> - output activation height</dd> |
| <dd><code>Q</code> - output activation width</dd> |
| <dd><code>intermediateMemoryBudget</code> - intermediate memory budget</dd> |
| </dl> |
| </li> |
| </ul> |
| <a name="conv2dBackwardData-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-int-int-int-int-int-int-int-int-int-int-int-int-int-double-"> |
| <!-- --> |
| </a> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <h4>conv2dBackwardData</h4> |
| <pre>public static void conv2dBackwardData(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> filter, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dout, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> output, |
| int N, |
| int C, |
| int H, |
| int W, |
| int K, |
| int R, |
| int S, |
| int pad_h, |
| int pad_w, |
| int stride_h, |
| int stride_w, |
| int P, |
| int Q, |
| double intermediateMemoryBudget)</pre> |
| <div class="block">This method computes the backpropogation errors for previous layer of convolution operation</div> |
| <dl> |
| <dt><span class="paramLabel">Parameters:</span></dt> |
| <dd><code>gCtx</code> - a valid <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context"><code>GPUContext</code></a></dd> |
| <dd><code>instName</code> - the invoking instruction's name for record <a href="../../../../../../org/apache/sysds/utils/Statistics.html" title="class in org.apache.sysds.utils"><code>Statistics</code></a>.</dd> |
| <dd><code>filter</code> - filter used in conv2d</dd> |
| <dd><code>dout</code> - errors from next layer</dd> |
| <dd><code>output</code> - output errors</dd> |
| <dd><code>N</code> - number of images</dd> |
| <dd><code>C</code> - number of channels</dd> |
| <dd><code>H</code> - height</dd> |
| <dd><code>W</code> - width</dd> |
| <dd><code>K</code> - number of filters</dd> |
| <dd><code>R</code> - filter height</dd> |
| <dd><code>S</code> - filter width</dd> |
| <dd><code>pad_h</code> - pad height</dd> |
| <dd><code>pad_w</code> - pad width</dd> |
| <dd><code>stride_h</code> - stride height</dd> |
| <dd><code>stride_w</code> - stride width</dd> |
| <dd><code>P</code> - output activation height</dd> |
| <dd><code>Q</code> - output activation width</dd> |
| <dd><code>intermediateMemoryBudget</code> - intermediate memory budget</dd> |
| </dl> |
| </li> |
| </ul> |
| <a name="pooling-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-int-int-int-int-int-int-int-int-int-int-int-int-int-org.apache.sysds.runtime.matrix.data.LibMatrixDNN.PoolingType-double-"> |
| <!-- --> |
| </a> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <h4>pooling</h4> |
| <pre>public static void pooling(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> outputBlock, |
| int N, |
| int C, |
| int H, |
| int W, |
| int K, |
| int R, |
| int S, |
| int pad_h, |
| int pad_w, |
| int stride_h, |
| int stride_w, |
| int P, |
| int Q, |
| <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixDNN.PoolingType.html" title="enum in org.apache.sysds.runtime.matrix.data">LibMatrixDNN.PoolingType</a> poolingType, |
| double intermediateMemoryBudget)</pre> |
| <div class="block">performs maxpooling on GPU by exploiting cudnnPoolingForward(...)</div> |
| <dl> |
| <dt><span class="paramLabel">Parameters:</span></dt> |
| <dd><code>gCtx</code> - a valid <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context"><code>GPUContext</code></a></dd> |
| <dd><code>instName</code> - the invoking instruction's name for record <a href="../../../../../../org/apache/sysds/utils/Statistics.html" title="class in org.apache.sysds.utils"><code>Statistics</code></a>.</dd> |
| <dd><code>image</code> - image as matrix object</dd> |
| <dd><code>outputBlock</code> - output matrix</dd> |
| <dd><code>N</code> - batch size</dd> |
| <dd><code>C</code> - number of channels</dd> |
| <dd><code>H</code> - height of image</dd> |
| <dd><code>W</code> - width of image</dd> |
| <dd><code>K</code> - number of filters</dd> |
| <dd><code>R</code> - height of filter</dd> |
| <dd><code>S</code> - width of filter</dd> |
| <dd><code>pad_h</code> - vertical padding</dd> |
| <dd><code>pad_w</code> - horizontal padding</dd> |
| <dd><code>stride_h</code> - horizontal stride</dd> |
| <dd><code>stride_w</code> - vertical stride</dd> |
| <dd><code>P</code> - (H - R + 1 + 2*pad_h)/stride_h</dd> |
| <dd><code>Q</code> - (W - S + 1 + 2*pad_w)/stride_w</dd> |
| <dd><code>poolingType</code> - type of pooling</dd> |
| <dd><code>intermediateMemoryBudget</code> - intermediate memory budget</dd> |
| </dl> |
| </li> |
| </ul> |
| <a name="poolingBackward-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-int-int-int-int-int-int-int-int-int-int-int-int-int-org.apache.sysds.runtime.matrix.data.LibMatrixDNN.PoolingType-double-"> |
| <!-- --> |
| </a> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <h4>poolingBackward</h4> |
| <pre>public static void poolingBackward(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dout, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> maxpoolOutput, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> outputBlock, |
| int N, |
| int C, |
| int H, |
| int W, |
| int K, |
| int R, |
| int S, |
| int pad_h, |
| int pad_w, |
| int stride_h, |
| int stride_w, |
| int P, |
| int Q, |
| <a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixDNN.PoolingType.html" title="enum in org.apache.sysds.runtime.matrix.data">LibMatrixDNN.PoolingType</a> poolingType, |
| double intermediateMemoryBudget)</pre> |
| <div class="block">Performs maxpoolingBackward on GPU by exploiting cudnnPoolingBackward(...) |
| This method computes the backpropogation errors for previous layer of maxpooling operation</div> |
| <dl> |
| <dt><span class="paramLabel">Parameters:</span></dt> |
| <dd><code>gCtx</code> - a valid <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context"><code>GPUContext</code></a></dd> |
| <dd><code>instName</code> - the invoking instruction's name for record <a href="../../../../../../org/apache/sysds/utils/Statistics.html" title="class in org.apache.sysds.utils"><code>Statistics</code></a>.</dd> |
| <dd><code>image</code> - image as matrix object</dd> |
| <dd><code>dout</code> - delta matrix, output of previous layer</dd> |
| <dd><code>maxpoolOutput</code> - (optional and can be null) output of maxpool forward function</dd> |
| <dd><code>outputBlock</code> - output matrix</dd> |
| <dd><code>N</code> - batch size</dd> |
| <dd><code>C</code> - number of channels</dd> |
| <dd><code>H</code> - height of image</dd> |
| <dd><code>W</code> - width of image</dd> |
| <dd><code>K</code> - number of filters</dd> |
| <dd><code>R</code> - height of filter</dd> |
| <dd><code>S</code> - width of filter</dd> |
| <dd><code>pad_h</code> - vertical padding</dd> |
| <dd><code>pad_w</code> - horizontal padding</dd> |
| <dd><code>stride_h</code> - horizontal stride</dd> |
| <dd><code>stride_w</code> - vertical stride</dd> |
| <dd><code>P</code> - (H - R + 1 + 2*pad_h)/stride_h</dd> |
| <dd><code>Q</code> - (W - S + 1 + 2*pad_w)/stride_w</dd> |
| <dd><code>poolingType</code> - type of pooling</dd> |
| <dd><code>intermediateMemoryBudget</code> - intermediate memory budget</dd> |
| </dl> |
| </li> |
| </ul> |
| <a name="relu-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-"> |
| <!-- --> |
| </a> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <h4>relu</h4> |
| <pre>public static void relu(<a href="../../../../../../org/apache/sysds/runtime/controlprogram/context/ExecutionContext.html" title="class in org.apache.sysds.runtime.controlprogram.context">ExecutionContext</a> ec, |
| <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> in, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> outputName)</pre> |
| <div class="block">Performs the relu operation on the GPU.</div> |
| <dl> |
| <dt><span class="paramLabel">Parameters:</span></dt> |
| <dd><code>ec</code> - currently active <a href="../../../../../../org/apache/sysds/runtime/controlprogram/context/ExecutionContext.html" title="class in org.apache.sysds.runtime.controlprogram.context"><code>ExecutionContext</code></a></dd> |
| <dd><code>gCtx</code> - a valid <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context"><code>GPUContext</code></a></dd> |
| <dd><code>instName</code> - the invoking instruction's name for record <a href="../../../../../../org/apache/sysds/utils/Statistics.html" title="class in org.apache.sysds.utils"><code>Statistics</code></a>.</dd> |
| <dd><code>in</code> - input matrix</dd> |
| <dd><code>outputName</code> - name of the output matrix</dd> |
| </dl> |
| </li> |
| </ul> |
| <a name="lstm-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-jcuda.Pointer-jcuda.Pointer-jcuda.Pointer-jcuda.Pointer-boolean-java.lang.String-java.lang.String-int-int-int-int-"> |
| <!-- --> |
| </a> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <h4>lstm</h4> |
| <pre>public static void lstm(<a href="../../../../../../org/apache/sysds/runtime/controlprogram/context/ExecutionContext.html" title="class in org.apache.sysds.runtime.controlprogram.context">ExecutionContext</a> ec, |
| <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| jcuda.Pointer X, |
| jcuda.Pointer wPointer, |
| jcuda.Pointer out0, |
| jcuda.Pointer c0, |
| boolean return_sequences, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> outputName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> cyName, |
| int N, |
| int M, |
| int D, |
| int T) |
| throws <a href="../../../../../../org/apache/sysds/runtime/DMLRuntimeException.html" title="class in org.apache.sysds.runtime">DMLRuntimeException</a></pre> |
| <div class="block">Computes the forward pass for an LSTM layer with M neurons. |
| The input data has N sequences of T examples, each with D features.</div> |
| <dl> |
| <dt><span class="paramLabel">Parameters:</span></dt> |
| <dd><code>ec</code> - execution context</dd> |
| <dd><code>gCtx</code> - gpu context</dd> |
| <dd><code>instName</code> - name of the instruction</dd> |
| <dd><code>X</code> - input matrix pointer</dd> |
| <dd><code>wPointer</code> - weight matrix pointer</dd> |
| <dd><code>out0</code> - Outputs from previous timestep</dd> |
| <dd><code>c0</code> - Initial cell state</dd> |
| <dd><code>return_sequences</code> - Whether to return `out` at all timesteps, or just for the final timestep.</dd> |
| <dd><code>outputName</code> - name of the out variable. If `return_sequences` is True, outputs for all timesteps.</dd> |
| <dd><code>cyName</code> - name of the output cell state. Cell state for final timestep.</dd> |
| <dd><code>N</code> - minibatch size</dd> |
| <dd><code>M</code> - hidden size</dd> |
| <dd><code>D</code> - number of features</dd> |
| <dd><code>T</code> - sequence length</dd> |
| <dt><span class="throwsLabel">Throws:</span></dt> |
| <dd><code><a href="../../../../../../org/apache/sysds/runtime/DMLRuntimeException.html" title="class in org.apache.sysds.runtime">DMLRuntimeException</a></code> - if error</dd> |
| </dl> |
| </li> |
| </ul> |
| <a name="lstmBackward-org.apache.sysds.runtime.controlprogram.context.ExecutionContext-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-jcuda.Pointer-jcuda.Pointer-jcuda.Pointer-jcuda.Pointer-java.lang.String-java.lang.String-java.lang.String-java.lang.String-java.lang.String-java.lang.String-java.lang.String-boolean-int-int-int-int-"> |
| <!-- --> |
| </a> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <h4>lstmBackward</h4> |
| <pre>public static void lstmBackward(<a href="../../../../../../org/apache/sysds/runtime/controlprogram/context/ExecutionContext.html" title="class in org.apache.sysds.runtime.controlprogram.context">ExecutionContext</a> ec, |
| <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| jcuda.Pointer x, |
| jcuda.Pointer hx, |
| jcuda.Pointer cx, |
| jcuda.Pointer wPointer, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> doutName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> dcyName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> dxName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> dwName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> dbName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> dhxName, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> dcxName, |
| boolean return_sequences, |
| int N, |
| int M, |
| int D, |
| int T) |
| throws <a href="../../../../../../org/apache/sysds/runtime/DMLRuntimeException.html" title="class in org.apache.sysds.runtime">DMLRuntimeException</a></pre> |
| <dl> |
| <dt><span class="throwsLabel">Throws:</span></dt> |
| <dd><code><a href="../../../../../../org/apache/sysds/runtime/DMLRuntimeException.html" title="class in org.apache.sysds.runtime">DMLRuntimeException</a></code></dd> |
| </dl> |
| </li> |
| </ul> |
| <a name="batchNormalizationForwardTraining-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-double-double-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-"> |
| <!-- --> |
| </a> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <h4>batchNormalizationForwardTraining</h4> |
| <pre>public static void batchNormalizationForwardTraining(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> scale, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> bias, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> runningMean, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> runningVar, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> ret, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> retRunningMean, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> retRunningVar, |
| double epsilon, |
| double exponentialAverageFactor, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> resultSaveMean, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> resultSaveInvVariance) |
| throws <a href="../../../../../../org/apache/sysds/runtime/DMLRuntimeException.html" title="class in org.apache.sysds.runtime">DMLRuntimeException</a></pre> |
| <div class="block">Performs the forward BatchNormalization layer computation for training</div> |
| <dl> |
| <dt><span class="paramLabel">Parameters:</span></dt> |
| <dd><code>gCtx</code> - a valid <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context"><code>GPUContext</code></a></dd> |
| <dd><code>instName</code> - name of the instruction</dd> |
| <dd><code>image</code> - input image</dd> |
| <dd><code>scale</code> - scale (as per CuDNN) and gamma as per original paper: shape [1, C, 1, 1]</dd> |
| <dd><code>bias</code> - bias (as per CuDNN) and beta as per original paper: shape [1, C, 1, 1]</dd> |
| <dd><code>runningMean</code> - running mean accumulated during training phase: shape [1, C, 1, 1]</dd> |
| <dd><code>runningVar</code> - running variance accumulated during training phase: shape [1, C, 1, 1]</dd> |
| <dd><code>ret</code> - (output) normalized input</dd> |
| <dd><code>retRunningMean</code> - (output) running mean accumulated during training phase: shape [1, C, 1, 1]</dd> |
| <dd><code>retRunningVar</code> - (output) running variance accumulated during training phase: shape [1, C, 1, 1]</dd> |
| <dd><code>epsilon</code> - epsilon value used in the batch normalization formula</dd> |
| <dd><code>exponentialAverageFactor</code> - factor used in the moving average computation</dd> |
| <dd><code>resultSaveMean</code> - (output) running mean accumulated during training phase: shape [1, C, 1, 1]</dd> |
| <dd><code>resultSaveInvVariance</code> - (output) running variance accumulated during training phase: shape [1, C, 1, 1]</dd> |
| <dt><span class="throwsLabel">Throws:</span></dt> |
| <dd><code><a href="../../../../../../org/apache/sysds/runtime/DMLRuntimeException.html" title="class in org.apache.sysds.runtime">DMLRuntimeException</a></code> - if error occurs</dd> |
| </dl> |
| </li> |
| </ul> |
| <a name="batchNormalizationForwardInference-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-double-"> |
| <!-- --> |
| </a> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <h4>batchNormalizationForwardInference</h4> |
| <pre>public static void batchNormalizationForwardInference(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> scale, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> bias, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> runningMean, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> runningVar, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> ret, |
| double epsilon) |
| throws <a href="../../../../../../org/apache/sysds/runtime/DMLRuntimeException.html" title="class in org.apache.sysds.runtime">DMLRuntimeException</a></pre> |
| <div class="block">Performs the forward BatchNormalization layer computation for inference</div> |
| <dl> |
| <dt><span class="paramLabel">Parameters:</span></dt> |
| <dd><code>gCtx</code> - a valid <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context"><code>GPUContext</code></a></dd> |
| <dd><code>instName</code> - name of the instruction</dd> |
| <dd><code>image</code> - input image</dd> |
| <dd><code>scale</code> - scale (as per CuDNN) and gamma as per original paper: shape [1, C, 1, 1]</dd> |
| <dd><code>bias</code> - bias (as per CuDNN) and beta as per original paper: shape [1, C, 1, 1]</dd> |
| <dd><code>runningMean</code> - running mean accumulated during training phase: shape [1, C, 1, 1]</dd> |
| <dd><code>runningVar</code> - running variance accumulated during training phase: shape [1, C, 1, 1]</dd> |
| <dd><code>ret</code> - normalized input</dd> |
| <dd><code>epsilon</code> - epsilon value used in the batch normalization formula</dd> |
| <dt><span class="throwsLabel">Throws:</span></dt> |
| <dd><code><a href="../../../../../../org/apache/sysds/runtime/DMLRuntimeException.html" title="class in org.apache.sysds.runtime">DMLRuntimeException</a></code> - if error occurs</dd> |
| </dl> |
| </li> |
| </ul> |
| <a name="batchNormalizationBackward-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-java.lang.String-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-double-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-"> |
| <!-- --> |
| </a> |
| <ul class="blockList"> |
| <li class="blockList"> |
| <h4>batchNormalizationBackward</h4> |
| <pre>public static void batchNormalizationBackward(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dout, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> scale, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dX, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dScale, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> dBias, |
| double epsilon, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> resultSaveMean, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> resultSaveInvVariance) |
| throws <a href="../../../../../../org/apache/sysds/runtime/DMLRuntimeException.html" title="class in org.apache.sysds.runtime">DMLRuntimeException</a></pre> |
| <div class="block">This method computes the backpropagation errors for image, scale and bias of batch normalization layer</div> |
| <dl> |
| <dt><span class="paramLabel">Parameters:</span></dt> |
| <dd><code>gCtx</code> - a valid <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context"><code>GPUContext</code></a></dd> |
| <dd><code>instName</code> - name of the instruction</dd> |
| <dd><code>image</code> - input image</dd> |
| <dd><code>dout</code> - input errors of shape C, H, W</dd> |
| <dd><code>scale</code> - scale (as per CuDNN) and gamma as per original paper: shape [1, C, 1, 1]</dd> |
| <dd><code>dX</code> - (output) backpropagation errors for previous layer</dd> |
| <dd><code>dScale</code> - backpropagation error for scale</dd> |
| <dd><code>dBias</code> - backpropagation error for bias</dd> |
| <dd><code>epsilon</code> - epsilon value used in the batch normalization formula</dd> |
| <dd><code>resultSaveMean</code> - (input) running mean accumulated during training phase: shape [1, C, 1, 1]</dd> |
| <dd><code>resultSaveInvVariance</code> - (input) running variance accumulated during training phase: shape [1, C, 1, 1]</dd> |
| <dt><span class="throwsLabel">Throws:</span></dt> |
| <dd><code><a href="../../../../../../org/apache/sysds/runtime/DMLRuntimeException.html" title="class in org.apache.sysds.runtime">DMLRuntimeException</a></code> - if error occurs</dd> |
| </dl> |
| </li> |
| </ul> |
| <a name="getDensePointerForCuDNN-org.apache.sysds.runtime.instructions.gpu.context.GPUContext-org.apache.sysds.runtime.controlprogram.caching.MatrixObject-java.lang.String-int-int-"> |
| <!-- --> |
| </a> |
| <ul class="blockListLast"> |
| <li class="blockList"> |
| <h4>getDensePointerForCuDNN</h4> |
| <pre>public static jcuda.Pointer getDensePointerForCuDNN(<a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context">GPUContext</a> gCtx, |
| <a href="../../../../../../org/apache/sysds/runtime/controlprogram/caching/MatrixObject.html" title="class in org.apache.sysds.runtime.controlprogram.caching">MatrixObject</a> image, |
| <a href="https://docs.oracle.com/javase/8/docs/api/java/lang/String.html?is-external=true" title="class or interface in java.lang">String</a> instName, |
| int numRows, |
| int numCols) |
| throws <a href="../../../../../../org/apache/sysds/runtime/DMLRuntimeException.html" title="class in org.apache.sysds.runtime">DMLRuntimeException</a></pre> |
| <div class="block">Convenience method to get jcudaDenseMatrixPtr. This method explicitly converts sparse to dense format, so use it judiciously.</div> |
| <dl> |
| <dt><span class="paramLabel">Parameters:</span></dt> |
| <dd><code>gCtx</code> - a valid <a href="../../../../../../org/apache/sysds/runtime/instructions/gpu/context/GPUContext.html" title="class in org.apache.sysds.runtime.instructions.gpu.context"><code>GPUContext</code></a></dd> |
| <dd><code>image</code> - input matrix object</dd> |
| <dd><code>instName</code> - name of the instruction</dd> |
| <dd><code>numRows</code> - expected number of rows</dd> |
| <dd><code>numCols</code> - expected number of columns</dd> |
| <dt><span class="returnLabel">Returns:</span></dt> |
| <dd>jcuda pointer</dd> |
| <dt><span class="throwsLabel">Throws:</span></dt> |
| <dd><code><a href="../../../../../../org/apache/sysds/runtime/DMLRuntimeException.html" title="class in org.apache.sysds.runtime">DMLRuntimeException</a></code> - if error occurs while sparse to dense conversion</dd> |
| </dl> |
| </li> |
| </ul> |
| </li> |
| </ul> |
| </li> |
| </ul> |
| </div> |
| </div> |
| <!-- ========= END OF CLASS DATA ========= --> |
| <!-- ======= START OF BOTTOM NAVBAR ====== --> |
| <div class="bottomNav"><a name="navbar.bottom"> |
| <!-- --> |
| </a> |
| <div class="skipNav"><a href="#skip.navbar.bottom" title="Skip navigation links">Skip navigation links</a></div> |
| <a name="navbar.bottom.firstrow"> |
| <!-- --> |
| </a> |
| <ul class="navList" title="Navigation"> |
| <li><a href="../../../../../../overview-summary.html">Overview</a></li> |
| <li><a href="package-summary.html">Package</a></li> |
| <li class="navBarCell1Rev">Class</li> |
| <li><a href="class-use/LibMatrixCuDNN.html">Use</a></li> |
| <li><a href="package-tree.html">Tree</a></li> |
| <li><a href="../../../../../../deprecated-list.html">Deprecated</a></li> |
| <li><a href="../../../../../../index-all.html">Index</a></li> |
| <li><a href="../../../../../../help-doc.html">Help</a></li> |
| </ul> |
| </div> |
| <div class="subNav"> |
| <ul class="navList"> |
| <li><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCUDA.html" title="class in org.apache.sysds.runtime.matrix.data"><span class="typeNameLink">Prev Class</span></a></li> |
| <li><a href="../../../../../../org/apache/sysds/runtime/matrix/data/LibMatrixCuDNNConvolutionAlgorithm.html" title="class in org.apache.sysds.runtime.matrix.data"><span class="typeNameLink">Next Class</span></a></li> |
| </ul> |
| <ul class="navList"> |
| <li><a href="../../../../../../index.html?org/apache/sysds/runtime/matrix/data/LibMatrixCuDNN.html" target="_top">Frames</a></li> |
| <li><a href="LibMatrixCuDNN.html" target="_top">No Frames</a></li> |
| </ul> |
| <ul class="navList" id="allclasses_navbar_bottom"> |
| <li><a href="../../../../../../allclasses-noframe.html">All Classes</a></li> |
| </ul> |
| <div> |
| <script type="text/javascript"><!-- |
| allClassesLink = document.getElementById("allclasses_navbar_bottom"); |
| if(window==top) { |
| allClassesLink.style.display = "block"; |
| } |
| else { |
| allClassesLink.style.display = "none"; |
| } |
| //--> |
| </script> |
| </div> |
| <div> |
| <ul class="subNavList"> |
| <li>Summary: </li> |
| <li>Nested | </li> |
| <li><a href="#fields.inherited.from.class.org.apache.sysds.runtime.matrix.data.LibMatrixCUDA">Field</a> | </li> |
| <li><a href="#constructor.summary">Constr</a> | </li> |
| <li><a href="#method.summary">Method</a></li> |
| </ul> |
| <ul class="subNavList"> |
| <li>Detail: </li> |
| <li>Field | </li> |
| <li><a href="#constructor.detail">Constr</a> | </li> |
| <li><a href="#method.detail">Method</a></li> |
| </ul> |
| </div> |
| <a name="skip.navbar.bottom"> |
| <!-- --> |
| </a></div> |
| <!-- ======== END OF BOTTOM NAVBAR ======= --> |
| <p class="legalCopy"><small>Copyright © 2021 <a href="https://www.apache.org/">The Apache Software Foundation</a>. All rights reserved.</small></p> |
| </body> |
| </html> |