Searched refs:splitGeneric (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Split.cpp | 31 bool splitGeneric(const Scalar* inputData, const Shape& inputShape, int32_t axis, in splitGeneric() function
61 return splitGeneric<_Float16>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat16()
68 return splitGeneric<float>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat32()
75 return splitGeneric<uint8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8()
82 return splitGeneric<int8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8Signed()
89 return splitGeneric<int32_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitInt32()
|