Navigation
API > API/Plugins > API/Plugins/NNE > API/Plugins/NNE/IModelInstanceCPU
References
| Module | NNE |
| Header | /Engine/Plugins/Experimental/NNE/Source/NNE/Public/NNERuntimeCPU.h |
| Include | #include "NNERuntimeCPU.h" |
int SetInputTensorShapes
&40;
TConstArrayView< FTensorShape > InInputShapes
&41;
Remarks
Prepare the model to be run with the given input shape.
The call is mandatory before a model can be run. The function will run shape inference and resolve, if possible, the output shapes which can then be accessed by calling GetOutputTensorShapes(). This is a potentially expensive call and should be called lazily if possible. 0 on success or a non-zero number otherwise.
Parameters
| Name | Description |
|---|---|
| InInputShapes | The input shapes to prepare the model with. |