Home
last modified time | relevance | path

Searched refs:splitsData (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/armnn/delegate/test/
H A DSplitTest.cpp143 std::vector<int32_t> splitsData { 3, 1 }; // Splits in SplitVUint8Test() local
165 splitsData, in SplitVUint8Test()
181 std::vector<int32_t> splitsData { 3, 1 }; // Splits in SplitVFp32Test() local
203 splitsData, in SplitVFp32Test()
H A DSplitTestHelper.hpp177 std::vector<int32_t>& splitsData, in CreateSplitVTfLiteModel() argument
189 … flatBufferBuilder.CreateVector(reinterpret_cast<const uint8_t*>(splitsData.data()), in CreateSplitVTfLiteModel()
190 sizeof(int32_t) * splitsData.size())); in CreateSplitVTfLiteModel()
286 std::vector<int32_t>& splitsData, in SplitVTest() argument
299 splitsData, in SplitVTest()
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/
H A DTfLiteParser.cpp4105 std::vector<int> splitsData(numSplits); in ParseSplitV() local
4107 ::memcpy(splitsData.data(), splitsBufferPtr->data.data(), splitsInfo.GetNumBytes()); in ParseSplitV()
4113 for (auto split : splitsData) in ParseSplitV()
4136 … splitsData[inferIdx] = armnn::numeric_cast<int>(inputTensorInfo.GetShape()[splitDim]) - splitSum; in ParseSplitV()
4152 unsigned int splitSize = armnn::numeric_cast<unsigned int>(splitsData[j]); in ParseSplitV()