Operations.h - OpenGrok cross reference for /frameworks/ml/nn/common/include/Operations.h

Lines Matching refs:Shape
44 struct Shape;
46 bool floorFloat16(const _Float16* inputData, _Float16* outputData, const Shape& shape);
47 bool floorFloat32(const float* inputData, float* outputData, const Shape& shape);
49 bool depthwiseConvFloat16(const _Float16* inputData, const Shape& inputShape,
50                           const _Float16* filterData, const Shape& filterShape,
51                           const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft,
55                           _Float16* outputData, const Shape& outputShape);
56 bool depthwiseConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData,
57                           const Shape& filterShape, const float* biasData, const Shape& biasShape,
62                           const Shape& outputShape);
63 bool depthwiseConvQuant8(const uint8_t* inputData, const Shape& inputShape,
64                          const uint8_t* filterData, const Shape& filterShape,
65                          const int32_t* biasData, const Shape& biasShape, int32_t paddingLeft,
69                          uint8_t* outputData, const Shape& outputShape);
70 bool depthwiseConvQuant8PerChannel(const uint8_t* inputData, const Shape& inputShape,
71                                    const int8_t* filterData, const Shape& filterShape,
73                                    const Shape& biasShape, int32_t paddingLeft,
78                                    const Shape& outputShape);
80 bool localResponseNormFloat16(const _Float16* inputData, const Shape& inputShape, int32_t radius,
82                               _Float16* outputData, const Shape& outputShape);
83 bool localResponseNormFloat32(const float* inputData, const Shape& inputShape, int32_t radius,
85                               const Shape& outputShape);
87 bool copyData(const void* inputData, const Shape& inputShape, void* outputData,
88               const Shape& outputShape);
91 bool depthToSpaceGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize,
92                          T* outputData, const Shape& outputShape);
94 bool spaceToDepthGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize,
95                          T* outputData, const Shape& outputShape);
98 bool padGeneric(const T* inputData, const Shape& inputShape, const int32_t* paddings, T pad_value,
99                 T* outputData, const Shape& outputShape);
102 bool batchToSpaceGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize,
103                          T* outputData, const Shape& outputShape);
106 bool spaceToBatchGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize,
107                          const int32_t* padding, const Shape& paddingShape, T* outputData,
108                          const Shape& outputShape);
110 bool meanFloat16(_Float16* inputData, const Shape& inputShape, const int32_t* axis,
111                  const Shape& axisShape, bool keepDims, _Float16* outputData,
112                  const Shape& outputShape);
114 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape,
115                  bool keepDims, T* outputData, const Shape& outputShape);
117 bool stridedSliceGeneric(const uint8_t* inputData, const Shape& inputShape,
120                          int32_t shrinkAxisMask, uint8_t* outputData, const Shape& outputShape);
122 bool argMinMaxGeneric(const uint8_t* inputData, const Shape& inputShape, int32_t axis,
123                       bool isArgMin, uint8_t* outputData, const Shape& outputShape);
125 bool splitFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis,
127                   const std::vector<Shape>& outputShapes);
129 bool splitFloat32(const float* inputData, const Shape& inputShape, const int32_t axis,
131                   const std::vector<Shape>& outputShapes);
133 bool splitInt32(const int32_t* inputData, const Shape& inputShape, const int32_t axis,
135                 const std::vector<Shape>& outputShapes);
137 bool splitQuant8(const uint8_t* inputData, const Shape& inputShape, const int32_t axis,
139                  const std::vector<Shape>& outputShapes);
141 bool groupedConvFloat16(const _Float16* inputData, const Shape& inputShape,
142                         const _Float16* filterData, const Shape& filterShape,
143                         const _Float16* biasData, const Shape& biasShape, int32_t numGroups,
146                         int32_t activation, _Float16* outputData, const Shape& outputShape);
148 bool groupedConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData,
149                         const Shape& filterShape, const float* biasData, const Shape& biasShape,
153                         const Shape& outputShape);
155 bool groupedConvQuant8(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData,
156                        const Shape& filterShape, const int32_t* biasData, const Shape& biasShape,
160                        const Shape& outputShape);
162 bool groupedConvQuant8PerChannel(const uint8_t* inputData, const Shape& inputShape,
163                                  const int8_t* filterData, const Shape& filterShape,
165                                  const Shape& biasShape, int32_t padding_left,
168                                  int32_t activation, uint8_t* outputData, const Shape& outputShape);
170 bool channelShuffleGeneric(const uint8_t* inputData, const Shape& inputShape, int32_t numGroups,
171                            int32_t axis, uint8_t* outputData, const Shape& outputShape);