Searched defs:axisData (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/armnn/delegate/test/ |
| H A D | SplitTest.cpp | 27 std::vector<int32_t> axisData { 3 }; // Axis in SplitUint8Test() local
57 std::vector<int32_t> axisData { 1 }; // Axis in SplitFp32Test() local
142 std::vector<int32_t> axisData { 1 }; // Axis in SplitVUint8Test() local
180 std::vector<int32_t> axisData { 1 }; // Axis in SplitVFp32Test() local
|
| H A D | SplitTestHelper.hpp | 28 std::vector<int32_t>& axisData, in CreateSplitTfLiteModel()
126 std::vector<int32_t>& axisData, in SplitTest()
178 std::vector<int32_t>& axisData, in CreateSplitVTfLiteModel()
287 std::vector<int32_t>& axisData, in SplitVTest()
|
| H A D | ReduceTestHelper.hpp | 29 std::vector<int32_t>& axisData, in CreateReduceTfLiteModel()
|
| H A D | ControlTestHelper.hpp | 120 std::vector<int32_t>& axisData, in CreateMeanTfLiteModel()
|
| /aosp_15_r20/external/armnn/src/armnnTfLiteParser/ |
| H A D | TfLiteParser.cpp | 2541 std::vector<int32_t> axisData(dimTensorInfo.GetNumElements()); in ParseMean() local
3950 std::vector<int32_t> axisData(axisTensorInfo.GetNumElements()); in ParseSplit() local
4070 std::vector<int> axisData(axisTensorInfo.GetNumElements()); in ParseSplitV() local
4229 std::vector<int32_t> axisData(axisTensorInfo.GetNumElements()); in ParseArgMinMax() local
4425 std::vector<int32_t> axisData(inputTensorInfo1.GetNumElements()); in ParseReduce() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/ |
| D | OperationsExecutionUtils.cpp | 501 bool meanPrepare(const Shape& input, const int32_t* axisData, const Shape& axisShape, bool keepDims, in meanPrepare()
|