• Home
  • Raw
  • Download

Lines Matching refs:Shape

53 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, const Shape& outputShape,  in reluFloat()
63 template bool reluFloat<float>(const float* inputData, const Shape& inputShape, float* outputData,
64 const Shape& outputShape, float reluMin, float reluMax);
65 template bool reluFloat<_Float16>(const _Float16* inputData, const Shape& inputShape,
66 _Float16* outputData, const Shape& outputShape, float reluMin,
70 bool relu1Float(const T* inputData, const Shape& inputShape, T* outputData, in relu1Float()
71 const Shape& outputShape) { in relu1Float()
74 template bool relu1Float<float>(const float* inputData, const Shape& inputShape, float* outputData,
75 const Shape& outputShape);
76 template bool relu1Float<_Float16>(const _Float16* inputData, const Shape& inputShape,
77 _Float16* outputData, const Shape& outputShape);
80 bool relu6Float(const T* inputData, const Shape& inputShape, T* outputData, in relu6Float()
81 const Shape& outputShape) { in relu6Float()
84 template bool relu6Float<float>(const float* inputData, const Shape& inputShape, float* outputData,
85 const Shape& outputShape);
86 template bool relu6Float<_Float16>(const _Float16* inputData, const Shape& inputShape,
87 _Float16* outputData, const Shape& outputShape);
89 bool tanhFloat16(const _Float16* inputData, const Shape& inputShape, _Float16* outputData, in tanhFloat16()
90 const Shape& outputShape) { in tanhFloat16()
99 bool tanhFloat32(const float* inputData, const Shape& inputShape, float* outputData, in tanhFloat32()
100 const Shape& outputShape) { in tanhFloat32()
110 bool logisticFloat(const T* inputData, const Shape& inputShape, T* outputData, in logisticFloat()
111 const Shape& outputShape) { in logisticFloat()
119 template bool logisticFloat<float>(const float* inputData, const Shape& inputShape,
120 float* outputData, const Shape& outputShape);
121 template bool logisticFloat<_Float16>(const _Float16* inputData, const Shape& inputShape,
122 _Float16* outputData, const Shape& outputShape);
125 inline bool reluXQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in reluXQuant8()
126 const Shape& outputShape) { in reluXQuant8()
141 bool reluQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in reluQuant8()
142 const Shape& outputShape) { in reluQuant8()
147 bool relu1Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu1Quant8()
148 const Shape& outputShape) { in relu1Quant8()
153 bool relu6Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu6Quant8()
154 const Shape& outputShape) { in relu6Quant8()
159 bool tanhQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in tanhQuant8()
160 const Shape& outputShape) { in tanhQuant8()
189 bool logisticQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in logisticQuant8()
190 const Shape& outputShape) { in logisticQuant8()
220 inline bool reluXQuant8Signed(const int8_t* inputData, const Shape& inputShape, int8_t* outputData, in reluXQuant8Signed()
221 const Shape& outputShape) { in reluXQuant8Signed()
236 bool reluQuant8Signed(const int8_t* inputData, const Shape& inputShape, int8_t* outputData, in reluQuant8Signed()
237 const Shape& outputShape) { in reluQuant8Signed()
242 bool relu1Quant8Signed(const int8_t* inputData, const Shape& inputShape, int8_t* outputData, in relu1Quant8Signed()
243 const Shape& outputShape) { in relu1Quant8Signed()
248 bool relu6Quant8Signed(const int8_t* inputData, const Shape& inputShape, int8_t* outputData, in relu6Quant8Signed()
249 const Shape& outputShape) { in relu6Quant8Signed()
254 bool tanhQuant8Signed(const int8_t* inputData, const Shape& inputShape, int8_t* outputData, in tanhQuant8Signed()
255 const Shape& outputShape) { in tanhQuant8Signed()
284 bool logisticQuant8Signed(const int8_t* inputData, const Shape& inputShape, int8_t* outputData, in logisticQuant8Signed()
285 const Shape& outputShape) { in logisticQuant8Signed()
328 bool hardSwishQuant(const T* inputData, const Shape& inputShape, T* outputData, in hardSwishQuant()
329 const Shape& outputShape) { in hardSwishQuant()
382 const Shape& input = context->getInputShape(kInputTensor); in validate()
410 Shape input = context->getInputShape(kInputTensor); in prepare()
414 Shape output = input; in prepare()
593 const Shape& inputShape = context->getInputShape(kInputTensor); in executeHardSwish()
594 const Shape& outputShape = context->getOutputShape(kOutputTensor); in executeHardSwish()