Lines Matching refs:outputShape

52 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, const Shape& outputShape,  in reluFloat()  argument
63 const Shape& outputShape, float reluMin, float reluMax);
65 _Float16* outputData, const Shape& outputShape, float reluMin,
70 const Shape& outputShape) { in relu1Float() argument
71 return reluFloat(inputData, inputShape, outputData, outputShape, -1.f, 1.f); in relu1Float()
74 const Shape& outputShape);
76 _Float16* outputData, const Shape& outputShape);
80 const Shape& outputShape) { in relu6Float() argument
81 return reluFloat(inputData, inputShape, outputData, outputShape, 0.f, 6.f); in relu6Float()
84 const Shape& outputShape);
86 _Float16* outputData, const Shape& outputShape);
89 const Shape& outputShape) { in tanhFloat16() argument
99 const Shape& outputShape) { in tanhFloat32() argument
110 const Shape& outputShape) { in logisticFloat() argument
119 float* outputData, const Shape& outputShape);
121 _Float16* outputData, const Shape& outputShape);
125 const Shape& outputShape) { in reluXQuant8() argument
141 const Shape& outputShape) { in reluQuant8() argument
143 return reluXQuant8<kActivationRelu>(inputData, inputShape, outputData, outputShape); in reluQuant8()
147 const Shape& outputShape) { in relu1Quant8() argument
149 return reluXQuant8<kActivationRelu1>(inputData, inputShape, outputData, outputShape); in relu1Quant8()
153 const Shape& outputShape) { in relu6Quant8() argument
155 return reluXQuant8<kActivationRelu6>(inputData, inputShape, outputData, outputShape); in relu6Quant8()
159 const Shape& outputShape) { in tanhQuant8() argument
161 if (outputShape.offset != 128 || outputShape.scale != 1.f / 128) { in tanhQuant8()
183 convertShapeToTflshape(outputShape)); in tanhQuant8()
189 const Shape& outputShape) { in logisticQuant8() argument
191 if (outputShape.offset != 0 || outputShape.scale != 1.f / 256) { in logisticQuant8()
213 input_multiplier, input_left_shift, outputData, convertShapeToTflshape(outputShape)); in logisticQuant8()
220 const Shape& outputShape) { in reluXQuant8Signed() argument
236 const Shape& outputShape) { in reluQuant8Signed() argument
238 return reluXQuant8Signed<kActivationRelu>(inputData, inputShape, outputData, outputShape); in reluQuant8Signed()
242 const Shape& outputShape) { in relu1Quant8Signed() argument
244 return reluXQuant8Signed<kActivationRelu1>(inputData, inputShape, outputData, outputShape); in relu1Quant8Signed()
248 const Shape& outputShape) { in relu6Quant8Signed() argument
250 return reluXQuant8Signed<kActivationRelu6>(inputData, inputShape, outputData, outputShape); in relu6Quant8Signed()
254 const Shape& outputShape) { in tanhQuant8Signed() argument
256 if (outputShape.offset != 0 || outputShape.scale != 1.f / 128) { in tanhQuant8Signed()
283 const Shape& outputShape) { in logisticQuant8Signed() argument
285 if (outputShape.offset != -128 || outputShape.scale != 1.f / 256) { in logisticQuant8Signed()
327 const Shape& outputShape) { in hardSwishQuant() argument
330 params.output_zero_point = outputShape.offset; in hardSwishQuant()
334 const float output_scale = outputShape.scale; in hardSwishQuant()
353 convertShapeToTflshape(outputShape), outputData); in hardSwishQuant()
410 auto outputShape = context->getOutputShape(kOutputTensor); in prepare() local
411 output.scale = outputShape.scale; in prepare()
412 output.offset = outputShape.offset; in prepare()
584 const Shape& outputShape = context->getOutputShape(kOutputTensor); in executeHardSwish() local
586 std::vector<float> outputFloat(getNumberOfElements(outputShape)); in executeHardSwish()
589 convertShapeToTflshape(outputShape), in executeHardSwish()