Home
last modified time | relevance | path

Searched defs:filterShape (Results 1 – 6 of 6) sorted by relevance

/packages/modules/NeuralNetworks/common/operations/
DConv2D.cpp108 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
178 bool needim2colData(const Shape& filterShape, int32_t stride_width, int32_t stride_height, in needim2colData()
195 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc()
224 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc()
278 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc()
307 const Shape& filterShape, const _Float16* biasData, const Shape& biasShape, in convNhwc()
334 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in conv()
353 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannelNhwc()
446 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannelNhwc()
510 const Shape& filterShape, const float* filterScales, in convQuant8PerChannel()
DTransposeConv2D.cpp75 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
127 const Shape& filterShape, const float* biasData, const Shape& biasShape, in transposeConvNhwc()
184 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc()
278 const _Float16* filterData, const Shape& filterShape, in transposeConvNhwc()
302 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in transposeConv()
318 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannelNhwc()
421 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannel()
DDepthwiseConv2D.cpp97 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
134 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvNhwc()
168 const _Float16* filterData, const Shape& filterShape, in depthwiseConvNhwc()
194 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc()
245 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc()
278 const Shape& filterShape, const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannelNhwc()
374 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in depthwiseConv()
395 const int8_t* filterData, const Shape& filterShape, in depthwiseConvQuant8PerChannel()
DGroupedConv2D.cpp47 const Shape& filterShape, const float* biasData, const Shape& biasShape, in groupedConvFloat32()
105 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8()
201 const int8_t* filterData, const Shape& filterShape, in groupedConvQuant8PerChannel()
286 const _Float16* filterData, const Shape& filterShape, in groupedConvFloat16()
/packages/modules/NeuralNetworks/common/
DOperationsUtils.cpp256 bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultipler()
DCpuExecutor.cpp1510 Shape filterShape = filter.shape(); in executeOperation() local