Operations.h - OpenGrok cross reference for /packages/modules/NeuralNetworks/common/include/Operations.h

Lines Matching refs:Shape
47 struct Shape;
49 bool floorFloat16(const _Float16* inputData, _Float16* outputData, const Shape& shape);
50 bool floorFloat32(const float* inputData, float* outputData, const Shape& shape);
52 bool depthwiseConvFloat16(const _Float16* inputData, const Shape& inputShape,
53                           const _Float16* filterData, const Shape& filterShape,
54                           const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft,
58                           _Float16* outputData, const Shape& outputShape);
59 bool depthwiseConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData,
60                           const Shape& filterShape, const float* biasData, const Shape& biasShape,
65                           const Shape& outputShape);
66 bool depthwiseConvQuant8(const uint8_t* inputData, const Shape& inputShape,
67                          const uint8_t* filterData, const Shape& filterShape,
68                          const int32_t* biasData, const Shape& biasShape, int32_t paddingLeft,
72                          uint8_t* outputData, const Shape& outputShape);
73 bool depthwiseConvQuant8PerChannel(const uint8_t* inputData, const Shape& inputShape,
74                                    const int8_t* filterData, const Shape& filterShape,
76                                    const Shape& biasShape, int32_t paddingLeft,
81                                    const Shape& outputShape);
83 bool localResponseNormFloat16(const _Float16* inputData, const Shape& inputShape, int32_t radius,
85                               _Float16* outputData, const Shape& outputShape);
86 bool localResponseNormFloat32(const float* inputData, const Shape& inputShape, int32_t radius,
88                               const Shape& outputShape);
90 bool copyData(const void* inputData, const Shape& inputShape, void* outputData,
91               const Shape& outputShape);
94 bool depthToSpaceGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize,
95                          T* outputData, const Shape& outputShape);
97 bool spaceToDepthGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize,
98                          T* outputData, const Shape& outputShape);
101 bool padGeneric(const T* inputData, const Shape& inputShape, const int32_t* paddings, T pad_value,
102                 T* outputData, const Shape& outputShape);
105 bool batchToSpaceGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize,
106                          T* outputData, const Shape& outputShape);
109 bool spaceToBatchGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize,
110                          const int32_t* padding, const Shape& paddingShape, T* outputData,
111                          const Shape& outputShape);
113 bool meanFloat16(_Float16* inputData, const Shape& inputShape, const int32_t* axis,
114                  const Shape& axisShape, bool keepDims, _Float16* outputData,
115                  const Shape& outputShape);
117 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape,
118                  bool keepDims, T* outputData, const Shape& outputShape);
120 bool stridedSliceGeneric(const uint8_t* inputData, const Shape& inputShape,
123                          int32_t shrinkAxisMask, uint8_t* outputData, const Shape& outputShape);
125 bool argMinMaxGeneric(const uint8_t* inputData, const Shape& inputShape, int32_t axis,
126                       bool isArgMin, uint8_t* outputData, const Shape& outputShape);
128 bool splitFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis,
130                   const std::vector<Shape>& outputShapes);
132 bool splitFloat32(const float* inputData, const Shape& inputShape, const int32_t axis,
134                   const std::vector<Shape>& outputShapes);
136 bool splitInt32(const int32_t* inputData, const Shape& inputShape, const int32_t axis,
138                 const std::vector<Shape>& outputShapes);
140 bool splitQuant8(const uint8_t* inputData, const Shape& inputShape, const int32_t axis,
142                  const std::vector<Shape>& outputShapes);
144 bool splitQuant8Signed(const int8_t* inputData, const Shape& inputShape, const int32_t axis,
146                        const std::vector<Shape>& outputShapes);
148 bool groupedConvFloat16(const _Float16* inputData, const Shape& inputShape,
149                         const _Float16* filterData, const Shape& filterShape,
150                         const _Float16* biasData, const Shape& biasShape, int32_t numGroups,
153                         int32_t activation, _Float16* outputData, const Shape& outputShape);
155 bool groupedConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData,
156                         const Shape& filterShape, const float* biasData, const Shape& biasShape,
160                         const Shape& outputShape);
163 bool groupedConvQuant8(const T* inputData, const Shape& inputShape, const T* filterData,
164                        const Shape& filterShape, const int32_t* biasData, const Shape& biasShape,
168                        const Shape& outputShape);
171 bool groupedConvQuant8PerChannel(const T* inputData, const Shape& inputShape,
172                                  const int8_t* filterData, const Shape& filterShape,
174                                  const Shape& biasShape, int32_t padding_left,
177                                  int32_t activation, T* outputData, const Shape& outputShape);
179 bool channelShuffleGeneric(const uint8_t* inputData, const Shape& inputShape, int32_t numGroups,
180                            int32_t axis, uint8_t* outputData, const Shape& outputShape);