Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DConv2D.cpp106 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
175 bool needim2colData(const Shape& filterShape, int32_t stride_width, int32_t stride_height, in needim2colData()
192 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc()
221 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc()
275 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc()
304 const Shape& filterShape, const _Float16* biasData, const Shape& biasShape, in convNhwc()
331 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in conv()
350 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannelNhwc()
443 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannelNhwc()
507 const Shape& filterShape, const float* filterScales, in convQuant8PerChannel()
DDepthwiseConv2D.cpp93 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
130 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvNhwc()
164 const _Float16* filterData, const Shape& filterShape, in depthwiseConvNhwc()
190 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc()
241 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc()
274 const Shape& filterShape, const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannelNhwc()
370 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in depthwiseConv()
391 const int8_t* filterData, const Shape& filterShape, in depthwiseConvQuant8PerChannel()
DTransposeConv2D.cpp70 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
123 const Shape& filterShape, const float* biasData, const Shape& /*biasShape*/, in transposeConvNhwc()
180 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc()
274 const _Float16* filterData, const Shape& filterShape, in transposeConvNhwc()
298 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in transposeConv()
314 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannelNhwc()
417 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannel()
DGroupedConv2D.cpp51 const Shape& filterShape, const float* biasData, const Shape& /*biasShape*/, in groupedConvFloat32()
109 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8()
205 const int8_t* filterData, const Shape& filterShape, in groupedConvQuant8PerChannel()
291 const _Float16* filterData, const Shape& filterShape, in groupedConvFloat16()
/packages/modules/NeuralNetworks/common/
DOperationsExecutionUtils.cpp154 bool GetQuantizedConvolutionMultiplier(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultiplier()
169 bool GetQuantizedConvolutionMultiplier(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultiplier()
DCpuExecutor.cpp1511 Shape filterShape = filter.shape(); in executeOperation() local