Lines Matching refs:Shape
96 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
97 Shape filterShape = context->getInputShape(kFilterTensor); in initialize()
133 bool depthwiseConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in depthwiseConvNhwc()
134 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvNhwc()
139 const Shape& outputShape) { in depthwiseConvNhwc()
167 bool depthwiseConvNhwc(const _Float16* inputData, const Shape& inputShape, in depthwiseConvNhwc()
168 const _Float16* filterData, const Shape& filterShape, in depthwiseConvNhwc()
169 const _Float16* biasData, const Shape& biasShape, int32_t paddingLeft, in depthwiseConvNhwc()
173 _Float16* outputData, const Shape& outputShape) { in depthwiseConvNhwc()
193 bool depthwiseConvNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in depthwiseConvNhwc()
194 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc()
199 const Shape& outputShape) { in depthwiseConvNhwc()
244 bool depthwiseConvNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in depthwiseConvNhwc()
245 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc()
250 Shape outputShape) { in depthwiseConvNhwc()
277 const T* inputData, const Shape& inputShape, const int8_t* filterData, in depthwiseConvQuant8PerChannelNhwc()
278 const Shape& filterShape, const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannelNhwc()
279 const Shape& biasShape, int32_t paddingLeft, int32_t paddingRight, int32_t paddingTop, in depthwiseConvQuant8PerChannelNhwc()
283 int32_t depthMultiplier, int32_t activation, T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannelNhwc()
308 Shape filterChannelShape = filterShape; in depthwiseConvQuant8PerChannelNhwc()
310 Shape biasChannelShape = biasShape; in depthwiseConvQuant8PerChannelNhwc()
373 bool depthwiseConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in depthwiseConv()
374 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in depthwiseConv()
379 const Shape& outputShape) { in depthwiseConv()
394 bool depthwiseConvQuant8PerChannel(const T* inputData, const Shape& inputShape, in depthwiseConvQuant8PerChannel()
395 const int8_t* filterData, const Shape& filterShape, in depthwiseConvQuant8PerChannel()
397 const Shape& biasShape, int32_t paddingLeft, in depthwiseConvQuant8PerChannel()
402 T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannel()
520 Shape input = context->getInputShape(kInputTensor); in prepare()
521 Shape filter = context->getInputShape(kFilterTensor); in prepare()
522 Shape bias = context->getInputShape(kBiasTensor); in prepare()
568 Shape output = context->getOutputShape(kOutputTensor); in prepare()