Searched defs:axisData (Results 1 – 5 of 5) sorted by relevance
27 std::vector<int32_t> axisData { 3 }; // Axis in SplitUint8Test() local57 std::vector<int32_t> axisData { 1 }; // Axis in SplitFp32Test() local142 std::vector<int32_t> axisData { 1 }; // Axis in SplitVUint8Test() local180 std::vector<int32_t> axisData { 1 }; // Axis in SplitVFp32Test() local
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()
29 std::vector<int32_t>& axisData, in CreateReduceTfLiteModel()
120 std::vector<int32_t>& axisData, in CreateMeanTfLiteModel()
2541 std::vector<int32_t> axisData(dimTensorInfo.GetNumElements()); in ParseMean() local3950 std::vector<int32_t> axisData(axisTensorInfo.GetNumElements()); in ParseSplit() local4070 std::vector<int> axisData(axisTensorInfo.GetNumElements()); in ParseSplitV() local4229 std::vector<int32_t> axisData(axisTensorInfo.GetNumElements()); in ParseArgMinMax() local4425 std::vector<int32_t> axisData(inputTensorInfo1.GetNumElements()); in ParseReduce() local