Update IOTensor.hpp

Signed-off-by: quic-zhanweiw <[email protected]>
quic · Sep 16, 2024 · af8f097 · af8f097
1 parent b00778a
commit af8f097
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/Utils/IOTensor.hpp b/src/Utils/IOTensor.hpp
@@ -76,6 +76,8 @@ class IOTensor {
 
   StatusCode fillDims(std::vector<size_t> &dims, uint32_t *inDimensions, uint32_t rank);	// zw: change it to public function.
 
+  StatusCode getTensorsSize(Qnn_Tensor_t** tensors, uint32_t tensorCount, Qnn_Tensor_t* tensorWrappers, std::vector<size_t>& size);
+
  private:
   PopulateInputTensorsRetType_t populateInputTensor(const std::vector<std::string> &filePaths,
                                                     const size_t filePathsIndexOffset,
@@ -114,7 +116,8 @@ class IOTensor {
   StatusCode copyFromFloatToNative(float *floatBuffer, Qnn_Tensor_t *tensor);
 
   StatusCode setupTensors(Qnn_Tensor_t **tensors, uint32_t tensorCount, Qnn_Tensor_t *tensorsInfo);
+
 };
 }  // namespace iotensor
 }  // namespace tools
-}  // namespace qnn
+}  // namespace qnn