static int |
mklml.dnnAllocateBuffer_F32(Pointer pPtr,
mklml._dnnLayout_s layout) |
static int |
mklml.dnnAllocateBuffer_F32(PointerPointer pPtr,
mklml._dnnLayout_s layout) |
static int |
mklml.dnnAllocateBuffer_F64(Pointer pPtr,
mklml._dnnLayout_s layout) |
static int |
mklml.dnnAllocateBuffer_F64(PointerPointer pPtr,
mklml._dnnLayout_s layout) |
static int |
mklml.dnnBatchNormalizationCreateBackward_v2_F32(mklml._uniPrimitive_s pBatchNormalization,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
float eps,
int flags) |
static int |
mklml.dnnBatchNormalizationCreateBackward_v2_F64(mklml._uniPrimitive_s pBatchNormalization,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
double eps,
int flags) |
static int |
mklml.dnnBatchNormalizationCreateBackwardData_F32(mklml._uniPrimitive_s pBatchNormalization,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
float eps) |
static int |
mklml.dnnBatchNormalizationCreateBackwardData_F64(mklml._uniPrimitive_s pBatchNormalization,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
double eps) |
static int |
mklml.dnnBatchNormalizationCreateBackwardScaleShift_F32(mklml._uniPrimitive_s pBatchNormalization,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
float eps) |
static int |
mklml.dnnBatchNormalizationCreateBackwardScaleShift_F64(mklml._uniPrimitive_s pBatchNormalization,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
double eps) |
static int |
mklml.dnnBatchNormalizationCreateForward_F32(mklml._uniPrimitive_s pBatchNormalization,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
float eps) |
static int |
mklml.dnnBatchNormalizationCreateForward_F64(mklml._uniPrimitive_s pBatchNormalization,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
double eps) |
static int |
mklml.dnnBatchNormalizationCreateForward_v2_F32(mklml._uniPrimitive_s pBatchNormalization,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
float eps,
int flags) |
static int |
mklml.dnnBatchNormalizationCreateForward_v2_F64(mklml._uniPrimitive_s pBatchNormalization,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
double eps,
int flags) |
static int |
mklml.dnnConcatCreate_F32(mklml._uniPrimitive_s pConcat,
mklml.dnnPrimitiveAttributes_t attributes,
long nSrcTensors,
mklml._dnnLayout_s src) |
static int |
mklml.dnnConcatCreate_F64(mklml._uniPrimitive_s pConcat,
mklml.dnnPrimitiveAttributes_t attributes,
long nSrcTensors,
mklml._dnnLayout_s src) |
static int |
mklml.dnnConversionCreate_F32(mklml._uniPrimitive_s pConversion,
mklml._dnnLayout_s from,
mklml._dnnLayout_s to) |
static int |
mklml.dnnConversionCreate_F64(mklml._uniPrimitive_s pConversion,
mklml._dnnLayout_s from,
mklml._dnnLayout_s to) |
static int |
mklml.dnnLayoutCompare_F32(mklml._dnnLayout_s l1,
mklml._dnnLayout_s l2) |
static int |
mklml.dnnLayoutCompare_F64(mklml._dnnLayout_s l1,
mklml._dnnLayout_s l2) |
static int |
mklml.dnnLayoutCreate_F32(mklml._dnnLayout_s pLayout,
long dimension,
SizeTPointer size,
SizeTPointer strides)
F32 section: single precision
|
static int |
mklml.dnnLayoutCreate_F64(mklml._dnnLayout_s pLayout,
long dimension,
SizeTPointer size,
SizeTPointer strides)
F64 section: double precision
|
static int |
mklml.dnnLayoutCreateFromPrimitive_F32(mklml._dnnLayout_s pLayout,
mklml._uniPrimitive_s primitive,
int type) |
static int |
mklml.dnnLayoutCreateFromPrimitive_F64(mklml._dnnLayout_s pLayout,
mklml._uniPrimitive_s primitive,
int type) |
static int |
mklml.dnnLayoutDelete_F32(mklml._dnnLayout_s layout) |
static int |
mklml.dnnLayoutDelete_F64(mklml._dnnLayout_s layout) |
static int |
mklml.dnnLayoutDeserialize_F32(mklml._dnnLayout_s pLayout,
Pointer buf)
Creates new layout restored from previously serialized one.
|
static int |
mklml.dnnLayoutDeserialize_F64(mklml._dnnLayout_s pLayout,
Pointer buf)
Creates new layout restored from previously serialized one.
|
static long |
mklml.dnnLayoutGetMemorySize_F32(mklml._dnnLayout_s layout) |
static long |
mklml.dnnLayoutGetMemorySize_F64(mklml._dnnLayout_s layout) |
static int |
mklml.dnnLayoutSerialize_F32(mklml._dnnLayout_s layout,
Pointer buf)
Serializes given \p layout into buffer \p buf.
|
static int |
mklml.dnnLayoutSerialize_F64(mklml._dnnLayout_s layout,
Pointer buf)
Serializes given \p layout into buffer \p buf.
|
static int |
mklml.dnnLRNCreateBackward_F32(mklml._uniPrimitive_s pLrn,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s diffLayout,
mklml._dnnLayout_s dataLayout,
long kernel_size,
float alpha,
float beta,
float k) |
static int |
mklml.dnnLRNCreateBackward_F64(mklml._uniPrimitive_s pLrn,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s diffLayout,
mklml._dnnLayout_s dataLayout,
long kernel_size,
double alpha,
double beta,
double k) |
static int |
mklml.dnnLRNCreateForward_F32(mklml._uniPrimitive_s pLrn,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
long kernel_size,
float alpha,
float beta,
float k) |
static int |
mklml.dnnLRNCreateForward_F64(mklml._uniPrimitive_s pLrn,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
long kernel_size,
double alpha,
double beta,
double k) |
static int |
mklml.dnnPoolingCreateBackward_F32(mklml._uniPrimitive_s pPooling,
mklml.dnnPrimitiveAttributes_t attributes,
int op,
mklml._dnnLayout_s srcLayout,
SizeTPointer kernelSize,
SizeTPointer kernelStride,
int[] inputOffset,
int borderType) |
static int |
mklml.dnnPoolingCreateBackward_F32(mklml._uniPrimitive_s pPooling,
mklml.dnnPrimitiveAttributes_t attributes,
int op,
mklml._dnnLayout_s srcLayout,
SizeTPointer kernelSize,
SizeTPointer kernelStride,
IntBuffer inputOffset,
int borderType) |
static int |
mklml.dnnPoolingCreateBackward_F32(mklml._uniPrimitive_s pPooling,
mklml.dnnPrimitiveAttributes_t attributes,
int op,
mklml._dnnLayout_s srcLayout,
SizeTPointer kernelSize,
SizeTPointer kernelStride,
IntPointer inputOffset,
int borderType) |
static int |
mklml.dnnPoolingCreateBackward_F64(mklml._uniPrimitive_s pPooling,
mklml.dnnPrimitiveAttributes_t attributes,
int op,
mklml._dnnLayout_s srcLayout,
SizeTPointer kernelSize,
SizeTPointer kernelStride,
int[] inputOffset,
int borderType) |
static int |
mklml.dnnPoolingCreateBackward_F64(mklml._uniPrimitive_s pPooling,
mklml.dnnPrimitiveAttributes_t attributes,
int op,
mklml._dnnLayout_s srcLayout,
SizeTPointer kernelSize,
SizeTPointer kernelStride,
IntBuffer inputOffset,
int borderType) |
static int |
mklml.dnnPoolingCreateBackward_F64(mklml._uniPrimitive_s pPooling,
mklml.dnnPrimitiveAttributes_t attributes,
int op,
mklml._dnnLayout_s srcLayout,
SizeTPointer kernelSize,
SizeTPointer kernelStride,
IntPointer inputOffset,
int borderType) |
static int |
mklml.dnnPoolingCreateForward_F32(mklml._uniPrimitive_s pPooling,
mklml.dnnPrimitiveAttributes_t attributes,
int op,
mklml._dnnLayout_s srcLayout,
SizeTPointer kernelSize,
SizeTPointer kernelStride,
int[] inputOffset,
int borderType) |
static int |
mklml.dnnPoolingCreateForward_F32(mklml._uniPrimitive_s pPooling,
mklml.dnnPrimitiveAttributes_t attributes,
int op,
mklml._dnnLayout_s srcLayout,
SizeTPointer kernelSize,
SizeTPointer kernelStride,
IntBuffer inputOffset,
int borderType) |
static int |
mklml.dnnPoolingCreateForward_F32(mklml._uniPrimitive_s pPooling,
mklml.dnnPrimitiveAttributes_t attributes,
int op,
mklml._dnnLayout_s srcLayout,
SizeTPointer kernelSize,
SizeTPointer kernelStride,
IntPointer inputOffset,
int borderType) |
static int |
mklml.dnnPoolingCreateForward_F64(mklml._uniPrimitive_s pPooling,
mklml.dnnPrimitiveAttributes_t attributes,
int op,
mklml._dnnLayout_s srcLayout,
SizeTPointer kernelSize,
SizeTPointer kernelStride,
int[] inputOffset,
int borderType) |
static int |
mklml.dnnPoolingCreateForward_F64(mklml._uniPrimitive_s pPooling,
mklml.dnnPrimitiveAttributes_t attributes,
int op,
mklml._dnnLayout_s srcLayout,
SizeTPointer kernelSize,
SizeTPointer kernelStride,
IntBuffer inputOffset,
int borderType) |
static int |
mklml.dnnPoolingCreateForward_F64(mklml._uniPrimitive_s pPooling,
mklml.dnnPrimitiveAttributes_t attributes,
int op,
mklml._dnnLayout_s srcLayout,
SizeTPointer kernelSize,
SizeTPointer kernelStride,
IntPointer inputOffset,
int borderType) |
static int |
mklml.dnnReLUCreateBackward_F32(mklml._uniPrimitive_s pRelu,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s diffLayout,
mklml._dnnLayout_s dataLayout,
float negativeSlope) |
static int |
mklml.dnnReLUCreateBackward_F64(mklml._uniPrimitive_s pRelu,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s diffLayout,
mklml._dnnLayout_s dataLayout,
double negativeSlope) |
static int |
mklml.dnnReLUCreateForward_F32(mklml._uniPrimitive_s pRelu,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
float negativeSlope) |
static int |
mklml.dnnReLUCreateForward_F64(mklml._uniPrimitive_s pRelu,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
double negativeSlope) |
static int |
mklml.dnnScaleCreate_F32(mklml._uniPrimitive_s pScale,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
float alpha) |
static int |
mklml.dnnScaleCreate_F64(mklml._uniPrimitive_s pScale,
mklml.dnnPrimitiveAttributes_t attributes,
mklml._dnnLayout_s dataLayout,
double alpha) |
static int |
mklml.dnnSplitCreate_F32(mklml._uniPrimitive_s pSplit,
mklml.dnnPrimitiveAttributes_t attributes,
long nDstTensors,
mklml._dnnLayout_s layout,
SizeTPointer dstChannelSize) |
static int |
mklml.dnnSplitCreate_F64(mklml._uniPrimitive_s pSplit,
mklml.dnnPrimitiveAttributes_t attributes,
long nDstTensors,
mklml._dnnLayout_s layout,
SizeTPointer dstChannelSize) |
static int |
mklml.dnnSumCreate_F32(mklml._uniPrimitive_s pSum,
mklml.dnnPrimitiveAttributes_t attributes,
long nSummands,
mklml._dnnLayout_s layout,
float[] coefficients) |
static int |
mklml.dnnSumCreate_F32(mklml._uniPrimitive_s pSum,
mklml.dnnPrimitiveAttributes_t attributes,
long nSummands,
mklml._dnnLayout_s layout,
FloatBuffer coefficients) |
static int |
mklml.dnnSumCreate_F32(mklml._uniPrimitive_s pSum,
mklml.dnnPrimitiveAttributes_t attributes,
long nSummands,
mklml._dnnLayout_s layout,
FloatPointer coefficients) |
static int |
mklml.dnnSumCreate_F64(mklml._uniPrimitive_s pSum,
mklml.dnnPrimitiveAttributes_t attributes,
long nSummands,
mklml._dnnLayout_s layout,
double[] coefficients) |
static int |
mklml.dnnSumCreate_F64(mklml._uniPrimitive_s pSum,
mklml.dnnPrimitiveAttributes_t attributes,
long nSummands,
mklml._dnnLayout_s layout,
DoubleBuffer coefficients) |
static int |
mklml.dnnSumCreate_F64(mklml._uniPrimitive_s pSum,
mklml.dnnPrimitiveAttributes_t attributes,
long nSummands,
mklml._dnnLayout_s layout,
DoublePointer coefficients) |