coder.getDeepLearningLayers

Get convolutional neural network layers supported for code generation for a specific deep learning library

collapse all in page

Syntax

coder.getDeepLearningLayers(libraryname)

Description

example

coder.getDeepLearningLayers(libraryname) returns the convolutional neural network layers supported for code generation for a specific deep learning library.

Note

To use coder.getDeepLearningLayers, you must install the support package that corresponds to libraryname:

  • For 'arm-compute' and 'mkldnn', install MATLAB® Coder™ Interface for Deep Learning Libraries.

  • For 'cudnn', 'tensorrt' or 'arm-compute-mali', install GPU Coder™ Interface for Deep Learning Libraries.

Examples

collapse all

Get a list of layers supported for code generation for Intel® Math Kernel Library for Deep Neural Networks.

coder.getDeepLearningLayers('mkldnn')

ans =

  17×1 cell array

    {'AdditionLayer'                 }
    {'AveragePooling2DLayer'         }
    {'BatchNormalizationLayer'       }
    {'ClassificationOutputLayer'     }
    {'ClippedReLULayer'              }
    {'Convolution2DLayer'            }
    {'CrossChannelNormalizationLayer'}
    {'DepthConcatenationLayer'       }
    {'DropoutLayer'                  }
    {'FullyConnectedLayer'           }
    {'ImageInputLayer'               }
    {'LeakyReLULayer'                }
    {'MaxPooling2DLayer'             }
    {'ReLULayer'                     }
    {'RegressionOutputLayer'         }
    {'SoftmaxLayer'                  }
    {'TransposedConvolution2DLayer'  }

Input Arguments

collapse all

Name of deep learning library, specified as one of the values in this table.

ValueDescription
'arm-compute'

ARM® Compute Library for targeting ARM CPU processors.

Requires the MATLAB Coder Interface for Deep Learning Libraries.

'arm-compute-mali'

ARM Compute Library for targeting ARM GPU processors.

Requires the GPU Coder product and the GPU Coder Interface for Deep Learning Libraries.

'cudnn'

cuDNN library.

Requires the GPU Coder product and the GPU Coder Interface for Deep Learning Libraries.

'mkldnn'

Intel Math Kernel Library for Deep Neural Networks.

Requires the MATLAB Coder Interface for Deep Learning Libraries.

'tensorrt'

TensorRT™ library.

Requires the GPU Coder product and the GPU Coder Interface for Deep Learning Libraries.

See Also

| |

Topics

Introduced in R2018b

GPU Coder Documentation
Support