Cross Reference: /frameworks/ml/nn/common/include/Operations.h

Lines Matching refs:inputShape
63 bool depthwiseConvFloat32(const float* inputData, const Shape& inputShape,
71 bool depthwiseConvQuant8(const uint8_t* inputData, const Shape& inputShape,
80 bool convFloat32(const float* inputData, const Shape& inputShape,
88 bool convQuant8(const uint8_t* inputData, const Shape& inputShape,
97 bool averagePoolFloat32(const float* inputData, const Shape& inputShape,
103 bool averagePoolQuant8(const uint8_t* inputData, const Shape& inputShape,
109 bool l2PoolFloat32(const float* inputData, const Shape& inputShape,
115 bool maxPoolFloat32(const float* inputData, const Shape& inputShape,
121 bool maxPoolQuant8(const uint8_t* inputData, const Shape& inputShape,
128 bool reluFloat32(const float* inputData, const Shape& inputShape,
130 bool relu1Float32(const float* inputData, const Shape& inputShape,
132 bool relu6Float32(const float* inputData, const Shape& inputShape,
134 bool tanhFloat32(const float* inputData, const Shape& inputShape,
136 bool logisticFloat32(const float* inputData, const Shape& inputShape,
138 bool softmaxFloat32(const float* inputData, const Shape& inputShape,
141 bool reluQuant8(const uint8_t* inputData, const Shape& inputShape,
143 bool relu1Quant8(const uint8_t* inputData, const Shape& inputShape,
145 bool relu6Quant8(const uint8_t* inputData, const Shape& inputShape,
147 bool logisticQuant8(const uint8_t* inputData, const Shape& inputShape,
149 bool softmaxQuant8(const uint8_t* inputData, const Shape& inputShape,
153 bool fullyConnectedFloat32(const float* inputData, const Shape& inputShape,
158 bool fullyConnectedQuant8(const uint8_t* inputData, const Shape& inputShape,
171 bool l2normFloat32(const float* inputData, const Shape& inputShape,
173 bool l2normQuant8(const uint8_t* inputData, const Shape& inputShape,
175 bool localResponseNormFloat32(const float* inputData, const Shape& inputShape,
179 bool reshapeGeneric(const void* inputData, const Shape& inputShape,
183                            const Shape& inputShape,
187 bool depthToSpaceGeneric(const uint8_t* inputData, const Shape& inputShape,
191 bool spaceToDepthGeneric(const uint8_t* inputData, const Shape& inputShape,
195 bool padGeneric(const uint8_t* inputData, const Shape& inputShape,
199 bool batchToSpaceGeneric(const uint8_t* inputData, const Shape& inputShape,
203 bool spaceToBatchGeneric(const uint8_t* inputData, const Shape& inputShape,
213 bool squeezeGeneric(const void* inputData, const Shape& inputShape,
221 bool transposeGeneric(const uint8_t* inputData, const Shape& inputShape,
225 bool meanGeneric(const uint8_t* inputData, const Shape& inputShape,
229 bool stridedSliceGeneric(const uint8_t* inputData, const Shape& inputShape,