Lines Matching refs:Shape

46 bool groupedConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData,  in groupedConvFloat32()
47 const Shape& filterShape, const float* biasData, const Shape& biasShape, in groupedConvFloat32()
51 const Shape& outputShape) { in groupedConvFloat32()
104 bool groupedConvQuant8(const T* inputData, const Shape& inputShape, const T* filterData, in groupedConvQuant8()
105 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8()
109 const Shape& outputShape) { in groupedConvQuant8()
181 template bool groupedConvQuant8<int8_t>(const int8_t* inputData, const Shape& inputShape,
182 const int8_t* filterData, const Shape& filterShape,
183 const int32_t* biasData, const Shape& biasShape,
188 const Shape& outputShape);
190 template bool groupedConvQuant8<uint8_t>(const uint8_t* inputData, const Shape& inputShape,
191 const uint8_t* filterData, const Shape& filterShape,
192 const int32_t* biasData, const Shape& biasShape,
197 const Shape& outputShape);
200 bool groupedConvQuant8PerChannel(const T* inputData, const Shape& inputShape, in groupedConvQuant8PerChannel()
201 const int8_t* filterData, const Shape& filterShape, in groupedConvQuant8PerChannel()
203 const Shape& biasShape, int32_t padding_left, in groupedConvQuant8PerChannel()
206 int32_t activation, T* outputData, const Shape& outputShape) { in groupedConvQuant8PerChannel()
218 Shape filterChannelShape = filterShape; in groupedConvQuant8PerChannel()
220 Shape biasChannelShape = biasShape; in groupedConvQuant8PerChannel()
285 bool groupedConvFloat16(const _Float16* inputData, const Shape& inputShape, in groupedConvFloat16()
286 const _Float16* filterData, const Shape& filterShape, in groupedConvFloat16()
287 const _Float16* biasData, const Shape& biasShape, int32_t padding_left, in groupedConvFloat16()
290 int32_t activation, _Float16* outputData, const Shape& outputShape) { in groupedConvFloat16()
312 const uint8_t* inputData, const Shape& inputShape, const int8_t* filterData,
313 const Shape& filterShape, const float* filterScales, const int32_t* biasData,
314 const Shape& biasShape, int32_t padding_left, int32_t padding_right, int32_t padding_top,
316 int32_t activation, uint8_t* outputData, const Shape& outputShape);
319 const int8_t* inputData, const Shape& inputShape, const int8_t* filterData,
320 const Shape& filterShape, const float* filterScales, const int32_t* biasData,
321 const Shape& biasShape, int32_t padding_left, int32_t padding_right, int32_t padding_top,
323 int32_t activation, int8_t* outputData, const Shape& outputShape);