diff options
author | Inki Dae <inki.dae@samsung.com> | 2020-05-29 15:15:57 +0900 |
---|---|---|
committer | Inki Dae <inki.dae@samsung.com> | 2020-05-29 15:15:57 +0900 |
commit | 500310cf792e74d386344369d354e11dffa862fd (patch) | |
tree | d1f5fe749cf323d279d6711aa3ef5b2e1b538056 | |
parent | 52dd12575f47b351d24ec9a69d4cd74f117a1d59 (diff) | |
download | mediavision-500310cf792e74d386344369d354e11dffa862fd.tar.gz mediavision-500310cf792e74d386344369d354e11dffa862fd.tar.bz2 mediavision-500310cf792e74d386344369d354e11dffa862fd.zip |
mv_inference: Add uint64 output tensor support
Change-Id: Ieda8df82ae27cf0120dd8a463bbed7aa436f900d
Signed-off-by: Inki Dae <inki.dae@samsung.com>
-rwxr-xr-x | mv_inference/inference/src/Inference.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/mv_inference/inference/src/Inference.cpp b/mv_inference/inference/src/Inference.cpp index 3b425563..c7b5d2f2 100755 --- a/mv_inference/inference/src/Inference.cpp +++ b/mv_inference/inference/src/Inference.cpp @@ -524,6 +524,8 @@ void Inference::CleanupTensorBuffers(void) delete[] (float *)tensor_buffer.buffer; else if (tensor_buffer.data_type == INFERENCE_TENSOR_DATA_TYPE_UINT32) delete[] (unsigned int *)tensor_buffer.buffer; + else if (tensor_buffer.data_type == INFERENCE_TENSOR_DATA_TYPE_UINT64) + delete[] (unsigned long long *)tensor_buffer.buffer; else delete[] (unsigned char *)tensor_buffer.buffer; } @@ -619,6 +621,9 @@ int Inference::PrepareTenosrBuffers(void) if (tensor_info.data_type == INFERENCE_TENSOR_DATA_TYPE_FLOAT32) { tensor_buffer.buffer = new float[tensor_info.size]; tensor_buffer.size = tensor_info.size * 4; + } else if (tensor_info.data_type == INFERENCE_TENSOR_DATA_TYPE_UINT64) { + tensor_buffer.buffer = new unsigned long long[tensor_info.size]; + tensor_buffer.size = tensor_info.size * 8; } else if (tensor_info.data_type == INFERENCE_TENSOR_DATA_TYPE_UINT32) { tensor_buffer.buffer = new unsigned int[tensor_info.size]; tensor_buffer.size = tensor_info.size * 4; |