Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DConv2D.cpp108 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
177 bool needim2colData(const Shape& filterShape, int32_t stride_width, int32_t stride_height, in needim2colData()
194 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc()
223 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc()
277 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc()
306 const Shape& filterShape, const _Float16* biasData, const Shape& biasShape, in convNhwc()
333 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in conv()
352 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannelNhwc()
445 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannelNhwc()
509 const Shape& filterShape, const float* filterScales, in convQuant8PerChannel()
DTransposeConv2D.cpp75 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
126 const Shape& filterShape, const float* biasData, const Shape& biasShape, in transposeConvNhwc()
183 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc()
277 const _Float16* filterData, const Shape& filterShape, in transposeConvNhwc()
301 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in transposeConv()
317 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannelNhwc()
420 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannel()
DDepthwiseConv2D.cpp95 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
132 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvNhwc()
166 const _Float16* filterData, const Shape& filterShape, in depthwiseConvNhwc()
192 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc()
243 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in depthwiseConvNhwc()
276 const Shape& filterShape, const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannelNhwc()
372 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in depthwiseConv()
393 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()
/frameworks/ml/nn/common/
DOperationsUtils.cpp258 bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultipler()
DCpuExecutor.cpp1566 Shape filterShape = filter.shape(); in executeOperation() local