Lines Matching refs:axis
29 bool splitGeneric(const Scalar* inputData, const Shape& inputShape, int32_t axis, in splitGeneric() argument
32 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in splitGeneric()
34 for (int i = 0; i < axis; ++i) { in splitGeneric()
39 for (int i = axis + 1; i < concatDimensions; ++i) { in splitGeneric()
46 const int copySize = outputShapes[i].dimensions[axis] * baseInnerSize; in splitGeneric()
55 bool splitFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in splitFloat16() argument
59 return splitGeneric<_Float16>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat16()
62 bool splitFloat32(const float* inputData, const Shape& inputShape, int32_t axis, in splitFloat32() argument
66 return splitGeneric<float>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat32()
69 bool splitQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in splitQuant8() argument
73 return splitGeneric<uint8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8()
76 bool splitQuant8Signed(const int8_t* inputData, const Shape& inputShape, int32_t axis, in splitQuant8Signed() argument
80 return splitGeneric<int8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8Signed()
83 bool splitInt32(const int32_t* inputData, const Shape& inputShape, int32_t axis, in splitInt32() argument
87 return splitGeneric<int32_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitInt32()