Lines Matching refs:inputShape
53 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, const Shape& outputShape, in reluFloat() argument
56 int numElements = getNumberOfElements(inputShape); in reluFloat()
63 template bool reluFloat<float>(const float* inputData, const Shape& inputShape, float* outputData,
65 template bool reluFloat<_Float16>(const _Float16* inputData, const Shape& inputShape,
70 bool relu1Float(const T* inputData, const Shape& inputShape, T* outputData, in relu1Float() argument
72 return reluFloat(inputData, inputShape, outputData, outputShape, -1.f, 1.f); in relu1Float()
74 template bool relu1Float<float>(const float* inputData, const Shape& inputShape, float* outputData,
76 template bool relu1Float<_Float16>(const _Float16* inputData, const Shape& inputShape,
80 bool relu6Float(const T* inputData, const Shape& inputShape, T* outputData, in relu6Float() argument
82 return reluFloat(inputData, inputShape, outputData, outputShape, 0.f, 6.f); in relu6Float()
84 template bool relu6Float<float>(const float* inputData, const Shape& inputShape, float* outputData,
86 template bool relu6Float<_Float16>(const _Float16* inputData, const Shape& inputShape,
89 bool tanhFloat16(const _Float16* inputData, const Shape& inputShape, _Float16* outputData, in tanhFloat16() argument
92 int numElements = getNumberOfElements(inputShape); in tanhFloat16()
99 bool tanhFloat32(const float* inputData, const Shape& inputShape, float* outputData, in tanhFloat32() argument
102 int numElements = getNumberOfElements(inputShape); in tanhFloat32()
110 bool logisticFloat(const T* inputData, const Shape& inputShape, T* outputData, in logisticFloat() argument
113 int numElements = getNumberOfElements(inputShape); in logisticFloat()
119 template bool logisticFloat<float>(const float* inputData, const Shape& inputShape,
121 template bool logisticFloat<_Float16>(const _Float16* inputData, const Shape& inputShape,
125 inline bool reluXQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in reluXQuant8() argument
127 int numElements = getNumberOfElements(inputShape); in reluXQuant8()
131 CalculateActivationRangeUint8(activation, inputShape, &output_activation_min, in reluXQuant8()
141 bool reluQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in reluQuant8() argument
144 return reluXQuant8<kActivationRelu>(inputData, inputShape, outputData, outputShape); in reluQuant8()
147 bool relu1Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu1Quant8() argument
150 return reluXQuant8<kActivationRelu1>(inputData, inputShape, outputData, outputShape); in relu1Quant8()
153 bool relu6Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu6Quant8() argument
156 return reluXQuant8<kActivationRelu6>(inputData, inputShape, outputData, outputShape); in relu6Quant8()
159 bool tanhQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in tanhQuant8() argument
167 int numElements = getNumberOfElements(inputShape); in tanhQuant8()
171 inputShape.scale * static_cast<double>(1 << (31 - kInputIntegerBits)); in tanhQuant8()
182 tflite::optimized_ops::Tanh(inputData, convertShapeToTflshape(inputShape), inputShape.offset, in tanhQuant8()
189 bool logisticQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in logisticQuant8() argument
197 int numElements = getNumberOfElements(inputShape); in logisticQuant8()
201 inputShape.scale * static_cast<double>(1 << (31 - kInputIntegerBits)); in logisticQuant8()
213 inputData, convertShapeToTflshape(inputShape), inputShape.offset, input_range_radius, in logisticQuant8()
220 inline bool reluXQuant8Signed(const int8_t* inputData, const Shape& inputShape, int8_t* outputData, in reluXQuant8Signed() argument
222 int numElements = getNumberOfElements(inputShape); in reluXQuant8Signed()
226 CalculateActivationRangeInt8(activation, inputShape, &output_activation_min, in reluXQuant8Signed()
236 bool reluQuant8Signed(const int8_t* inputData, const Shape& inputShape, int8_t* outputData, in reluQuant8Signed() argument
239 return reluXQuant8Signed<kActivationRelu>(inputData, inputShape, outputData, outputShape); in reluQuant8Signed()
242 bool relu1Quant8Signed(const int8_t* inputData, const Shape& inputShape, int8_t* outputData, in relu1Quant8Signed() argument
245 return reluXQuant8Signed<kActivationRelu1>(inputData, inputShape, outputData, outputShape); in relu1Quant8Signed()
248 bool relu6Quant8Signed(const int8_t* inputData, const Shape& inputShape, int8_t* outputData, in relu6Quant8Signed() argument
251 return reluXQuant8Signed<kActivationRelu6>(inputData, inputShape, outputData, outputShape); in relu6Quant8Signed()
254 bool tanhQuant8Signed(const int8_t* inputData, const Shape& inputShape, int8_t* outputData, in tanhQuant8Signed() argument
262 int numElements = getNumberOfElements(inputShape); in tanhQuant8Signed()
266 inputShape.scale * static_cast<double>(1 << (31 - kInputIntegerBits)); in tanhQuant8Signed()
277 tflite::reference_integer_ops::Tanh(inputShape.offset, input_range_radius, input_multiplier, in tanhQuant8Signed()
278 input_left_shift, convertShapeToTflshape(inputShape), in tanhQuant8Signed()
284 bool logisticQuant8Signed(const int8_t* inputData, const Shape& inputShape, int8_t* outputData, in logisticQuant8Signed() argument
292 int numElements = getNumberOfElements(inputShape); in logisticQuant8Signed()
296 inputShape.scale * static_cast<double>(1 << (31 - kInputIntegerBits)); in logisticQuant8Signed()
307 tflite::reference_integer_ops::Logistic(inputShape.offset, input_range_radius, input_multiplier, in logisticQuant8Signed()
328 bool hardSwishQuant(const T* inputData, const Shape& inputShape, T* outputData, in hardSwishQuant() argument
331 params.input_zero_point = inputShape.offset; in hardSwishQuant()
333 const float input_scale = inputShape.scale; in hardSwishQuant()
354 tflite::reference_ops::HardSwish(params, convertShapeToTflshape(inputShape), inputData, in hardSwishQuant()
593 const Shape& inputShape = context->getInputShape(kInputTensor); in executeHardSwish() local
595 std::vector<float> inputFloat(getNumberOfElements(inputShape)); in executeHardSwish()
598 tflite::reference_ops::HardSwish(convertShapeToTflshape(inputShape), inputFloat.data(), in executeHardSwish()