Lines Matching refs:outputDataPtrs
31 const std::vector<Scalar*>* outputDataPtrs, in splitGeneric() argument
46 for (int i = 0; i < outputDataPtrs->size(); ++i) { in splitGeneric()
48 memcpy(outputDataPtrs->at(i) + k * copySize, inputPtr, copySize * sizeof(Scalar)); in splitGeneric()
57 const std::vector<_Float16*>* outputDataPtrs, in splitFloat16() argument
60 return splitGeneric<_Float16>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat16()
64 const std::vector<float*>* outputDataPtrs, in splitFloat32() argument
67 return splitGeneric<float>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat32()
71 const std::vector<uint8_t*>* outputDataPtrs, in splitQuant8() argument
74 return splitGeneric<uint8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8()
78 const std::vector<int8_t*>* outputDataPtrs, in splitQuant8Signed() argument
81 return splitGeneric<int8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8Signed()
85 const std::vector<int32_t*>* outputDataPtrs, in splitInt32() argument
88 return splitGeneric<int32_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitInt32()