Split.cpp - OpenGrok cross reference for /aosp12/packages/modules/NeuralNetworks/common/operations/Split.cpp

Lines Matching refs:Shape

29 bool splitGeneric(const Scalar* inputData, const Shape& inputShape, int32_t axis,  in splitGeneric()
31                   const std::vector<Shape>& outputShapes) {  in splitGeneric()
55 bool splitFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis,  in splitFloat16()
57                   const std::vector<Shape>& outputShapes) {  in splitFloat16()
62 bool splitFloat32(const float* inputData, const Shape& inputShape, int32_t axis,  in splitFloat32()
64                   const std::vector<Shape>& outputShapes) {  in splitFloat32()
69 bool splitQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis,  in splitQuant8()
71                  const std::vector<Shape>& outputShapes) {  in splitQuant8()
76 bool splitQuant8Signed(const int8_t* inputData, const Shape& inputShape, int32_t axis,  in splitQuant8Signed()
78                        const std::vector<Shape>& outputShapes) {  in splitQuant8Signed()
83 bool splitInt32(const int32_t* inputData, const Shape& inputShape, int32_t axis,  in splitInt32()
85                 const std::vector<Shape>& outputShapes) {  in splitInt32()