Function Snpe_SNPEBuilder_SetInputDimensionsForNetwork

• Defined in File SNPEBuilder.h

Function Documentation

Snpe_ErrorCode_t Snpe_SNPEBuilder_SetInputDimensionsForNetwork(Snpe_SNPEBuilder_Handle_t snpeBuilderHandle, const char *networkName, Snpe_TensorShapeMap_Handle_t inputDimensionsMapHandle)

Sets network’s input dimensions to enable resizing of the spatial dimensions of each layer for fully convolutional networks, and the batch dimension for all networks.

Parameters

• networkName – [in] : specifies network name on which the input dims are to be set

• tensorShapeMapHandle – [in] : Handle to the map of input names and their new dimensions. The new dimensions overwrite the input dimensions embedded in the model and then resize each layer of the model. If the model contains layers whose dimensions cannot be resized e.g FullyConnected, exception will be thrown when SNPE instance is actually built. In general the batch dimension is always resizable. After resizing of layers’ dimensions in model based on new input dimensions, the new model is revalidated against all runtime constraints, whose failures may result in cpu fallback situation.