Lines Matching refs:inputShape
29 bool splitGeneric(const Scalar* inputData, const Shape& inputShape, int32_t axis, in splitGeneric() argument
32 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in splitGeneric()
35 outerSize *= inputShape.dimensions[i]; in splitGeneric()
38 int concatDimensions = getNumberOfDimensions(inputShape); in splitGeneric()
40 baseInnerSize *= inputShape.dimensions[i]; 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()