Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DTransposeConv2D.cpp65 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
116 const Shape& filterShape, const float* biasData, const Shape& biasShape, in transposeConvNhwc()
172 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc()
267 const _Float16* filterData, const Shape& filterShape, in transposeConvNhwc()
291 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in transposeConv()
306 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannelNhwc()
408 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannel()
DDepthwiseConv2D.cpp29 const _Float16* filterData, const Shape& filterShape, in depthwiseConvFloat16()
66 const Shape& filterShape, const float* biasData, const Shape& biasShape, in depthwiseConvFloat32()
99 const uint8_t* filterData, const Shape& filterShape, in depthwiseConvQuant8()
147 const int8_t* filterData, const Shape& filterShape, in depthwiseConvQuant8PerChannel()
DConv2D.cpp95 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local
165 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc()
190 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc()
237 const Shape& filterShape, const _Float16* biasData, const Shape& biasShape, in convNhwc()
264 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in conv()
283 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannelNhwc()
376 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannel()
DGroupedConv2D.cpp43 const Shape& filterShape, const float* biasData, const Shape& biasShape, in groupedConvFloat32()
100 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in groupedConvQuant8()
177 const int8_t* filterData, const Shape& filterShape, in groupedConvQuant8PerChannel()
263 const _Float16* filterData, const Shape& filterShape, in groupedConvFloat16()
/frameworks/ml/nn/common/
DCpuExecutor.cpp810 Shape filterShape = filter.shape(); in executeOperation() local
1595 Shape filterShape = filter.shape(); in executeOperation() local
DOperationsUtils.cpp213 bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultipler()