Searched refs:depthMultiplier (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | DepthwiseConv2D.cpp | 33 int32_t dilationHeightFactor, int32_t depthMultiplier, int32_t activation, in depthwiseConvFloat16() argument
47 dilationHeightFactor, depthMultiplier, activation, in depthwiseConvFloat16()
70 int32_t depthMultiplier, int32_t activation, float* outputData, in depthwiseConvFloat32() argument
83 .depth_multiplier = static_cast<int16>(depthMultiplier), in depthwiseConvFloat32()
103 int32_t dilationHeightFactor, int32_t depthMultiplier, int32_t activation, in depthwiseConvQuant8() argument
127 .depth_multiplier = static_cast<int16>(depthMultiplier), in depthwiseConvQuant8()
154 int32_t depthMultiplier, int32_t activation, uint8_t* outputData, in depthwiseConvQuant8PerChannel() argument
201 for (uint32_t m = 0; m < depthMultiplier; m++) { in depthwiseConvQuant8PerChannel()
204 const int oc = m + ic * depthMultiplier; in depthwiseConvQuant8PerChannel()
234 outPtr += depthMultiplier; in depthwiseConvQuant8PerChannel()
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 54 int32_t dilationHeightFactor, int32_t depthMultiplier, int32_t activation,
61 int32_t depthMultiplier, int32_t activation, float* outputData,
68 int32_t dilationHeightFactor, int32_t depthMultiplier, int32_t activation,
77 int32_t depthMultiplier, int32_t activation, uint8_t* outputData,
|