Lines Matching refs:outputShapes
32 const std::vector<Shape>& outputShapes) { in splitGeneric() argument
47 const int copySize = outputShapes[i].dimensions[axis] * baseInnerSize; in splitGeneric()
58 const std::vector<Shape>& outputShapes) { in splitFloat16() argument
60 return splitGeneric<_Float16>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat16()
65 const std::vector<Shape>& outputShapes) { in splitFloat32() argument
67 return splitGeneric<float>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat32()
72 const std::vector<Shape>& outputShapes) { in splitQuant8() argument
74 return splitGeneric<uint8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8()
79 const std::vector<Shape>& outputShapes) { in splitQuant8Signed() argument
81 return splitGeneric<int8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8Signed()
86 const std::vector<Shape>& outputShapes) { in splitInt32() argument
88 return splitGeneric<int32_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitInt32()