public class AggregateGEMM extends BaseAggregate
arguments, finalResult, indexingArguments, intArrayArguments, realArguments, shapes
Constructor and Description |
---|
AggregateGEMM() |
AggregateGEMM(int Order,
int TransA,
int TransB,
int M,
int N,
int K,
double alpha,
INDArray A,
int lda,
INDArray B,
int ldb,
double beta,
INDArray C,
int ldc) |
Modifier and Type | Method and Description |
---|---|
int |
getSharedMemorySize()
This method returns amount of shared memory required for this specific Aggregate.
|
int |
getThreadsPerInstance()
This method returns desired number of threads per Aggregate instance
PLEASE NOTE: this method is especially important for CUDA backend.
|
int |
maxArguments()
This method returns maximum number of shapes being passed per Aggregate
|
int |
maxIndexArguments()
This method returns maximum number of IndexArguments per Aggregate
|
int |
maxIntArrays()
This method returns maximum number of IntArrays being passed per Aggregate
|
int |
maxIntArraySize()
This method returns maximum length for IntArrays, if any
|
int |
maxRealArguments()
This method returns maximum number of real (float/double) per Aggregate
|
int |
maxShapes()
This method returns maximum number of shapes being passed per Aggregate
|
String |
name() |
int |
opNum() |
getArguments, getFinalResult, getIndexingArguments, getIntArrayArguments, getRealArguments, getRequiredBatchMemorySize, getShapes, setFinalResult
public String name()
public int opNum()
public int maxArguments()
Aggregate
public int maxShapes()
Aggregate
public int maxIntArrays()
Aggregate
public int maxIntArraySize()
Aggregate
public int maxIndexArguments()
Aggregate
public int maxRealArguments()
Aggregate
public int getSharedMemorySize()
Aggregate
public int getThreadsPerInstance()
Aggregate
Copyright © 2017. All rights reserved.