Lines Matching refs:outputShape

53 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, const Shape& outputShape,  in reluFloat()  argument
64 const Shape& outputShape, float reluMin, float reluMax);
66 _Float16* outputData, const Shape& outputShape, float reluMin,
71 const Shape& outputShape) { in relu1Float() argument
72 return reluFloat(inputData, inputShape, outputData, outputShape, -1.f, 1.f); in relu1Float()
75 const Shape& outputShape);
77 _Float16* outputData, const Shape& outputShape);
81 const Shape& outputShape) { in relu6Float() argument
82 return reluFloat(inputData, inputShape, outputData, outputShape, 0.f, 6.f); in relu6Float()
85 const Shape& outputShape);
87 _Float16* outputData, const Shape& outputShape);
90 const Shape& outputShape) { in tanhFloat16() argument
100 const Shape& outputShape) { in tanhFloat32() argument
111 const Shape& outputShape) { in logisticFloat() argument
120 float* outputData, const Shape& outputShape);
122 _Float16* outputData, const Shape& outputShape);
126 const Shape& outputShape) { in reluXQuant8() argument
142 const Shape& outputShape) { in reluQuant8() argument
144 return reluXQuant8<kActivationRelu>(inputData, inputShape, outputData, outputShape); in reluQuant8()
148 const Shape& outputShape) { in relu1Quant8() argument
150 return reluXQuant8<kActivationRelu1>(inputData, inputShape, outputData, outputShape); in relu1Quant8()
154 const Shape& outputShape) { in relu6Quant8() argument
156 return reluXQuant8<kActivationRelu6>(inputData, inputShape, outputData, outputShape); in relu6Quant8()
160 const Shape& outputShape) { in tanhQuant8() argument
162 if (outputShape.offset != 128 || outputShape.scale != 1.f / 128) { in tanhQuant8()
184 convertShapeToTflshape(outputShape)); in tanhQuant8()
190 const Shape& outputShape) { in logisticQuant8() argument
192 if (outputShape.offset != 0 || outputShape.scale != 1.f / 256) { in logisticQuant8()
214 input_multiplier, input_left_shift, outputData, convertShapeToTflshape(outputShape)); in logisticQuant8()
221 const Shape& outputShape) { in reluXQuant8Signed() argument
237 const Shape& outputShape) { in reluQuant8Signed() argument
239 return reluXQuant8Signed<kActivationRelu>(inputData, inputShape, outputData, outputShape); in reluQuant8Signed()
243 const Shape& outputShape) { in relu1Quant8Signed() argument
245 return reluXQuant8Signed<kActivationRelu1>(inputData, inputShape, outputData, outputShape); in relu1Quant8Signed()
249 const Shape& outputShape) { in relu6Quant8Signed() argument
251 return reluXQuant8Signed<kActivationRelu6>(inputData, inputShape, outputData, outputShape); in relu6Quant8Signed()
255 const Shape& outputShape) { in tanhQuant8Signed() argument
257 if (outputShape.offset != 0 || outputShape.scale != 1.f / 128) { in tanhQuant8Signed()
279 inputData, convertShapeToTflshape(outputShape), outputData); in tanhQuant8Signed()
285 const Shape& outputShape) { in logisticQuant8Signed() argument
287 if (outputShape.offset != -128 || outputShape.scale != 1.f / 256) { in logisticQuant8Signed()
329 const Shape& outputShape) { in hardSwishQuant() argument
332 params.output_zero_point = outputShape.offset; in hardSwishQuant()
336 const float output_scale = outputShape.scale; in hardSwishQuant()
355 convertShapeToTflshape(outputShape), outputData); in hardSwishQuant()
420 auto outputShape = context->getOutputShape(kOutputTensor); in prepare() local
421 output.scale = outputShape.scale; in prepare()
422 output.offset = outputShape.offset; in prepare()
594 const Shape& outputShape = context->getOutputShape(kOutputTensor); in executeHardSwish() local
596 std::vector<float> outputFloat(getNumberOfElements(outputShape)); in executeHardSwish()
599 convertShapeToTflshape(outputShape), in executeHardSwish()