Searched refs:imageTensorFile (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/armnn/tests/ImageTensorGenerator/ |
| H A D | ImageTensorGenerator.cpp | 330 std::ofstream imageTensorFile; in main() local
331 imageTensorFile.open(outputPath, std::ofstream::out); in main()
332 if (imageTensorFile.is_open()) in main()
335 … [&imageTensorFile](auto&& imageData){ WriteImageTensorImpl(imageData,imageTensorFile); }, in main()
339 if (!imageTensorFile) in main()
342 imageTensorFile.close(); in main()
345 imageTensorFile.close(); in main()
|
| H A D | ImageTensorGenerator.hpp | 192 void WriteImageTensorImpl(const std::vector<ElemType>& imageData, std::ofstream& imageTensorFile) in WriteImageTensorImpl() argument
194 …std::copy(imageData.begin(), imageData.end(), std::ostream_iterator<ElemType>(imageTensorFile, " "… in WriteImageTensorImpl()
200 …riteImageTensorImpl<uint8_t>(const std::vector<uint8_t>& imageData, std::ofstream& imageTensorFile) in WriteImageTensorImpl() argument
202 std::copy(imageData.begin(), imageData.end(), std::ostream_iterator<int>(imageTensorFile, " ")); in WriteImageTensorImpl()
208 … WriteImageTensorImpl<int8_t>(const std::vector<int8_t>& imageData, std::ofstream& imageTensorFile) in WriteImageTensorImpl() argument
210 std::copy(imageData.begin(), imageData.end(), std::ostream_iterator<int>(imageTensorFile, " ")); in WriteImageTensorImpl()
|