1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
|
float input[30];
float output[30];
// resize input tensor
int inputTensorSize = 30;
int inputDims[3] = { 1, inputTensorSize, 1 };
TfLiteInterpreterResizeInputTensor(IA_config0->interpreter_direct, 0, inputDims, 3);
// re-allocate tensors
TfLiteInterpreterAllocateTensors(IA_config0->interpreter_direct);
// input buffer to input tensor
IA_config0->input_tensor_direct = TfLiteInterpreterGetInputTensor(IA_config0->interpreter_direct, 0);
///////////////////////////////////////////////////////////////////
statut = TfLiteTensorCopyFromBuffer(IA_config0->input_tensor_direct, input, inputTensorSize * sizeof(float));
// Execute inference.
statut = TfLiteInterpreterInvoke(IA_config0->interpreter_direct);
// Extract the output tensor data.
statut = TfLiteTensorCopyToBuffer(IA_config0->output_tensor_direct, output, TfLiteTensorByteSize(IA_config0->output_tensor_direct)); |
Partager