Searched defs:filterShape (Results 1 – 5 of 5) sorted by relevance
/frameworks/ml/nn/common/operations/ |
D | TransposeConv2D.cpp | 75 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()
|
D | Conv2D.cpp | 108 Shape filterShape = context->getInputShape(kFilterTensor); in initialize() local 178 const Shape& filterShape, const float* biasData, const Shape& biasShape, in convNhwc() 203 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc() 252 Shape filterShape, const int32_t* biasData, const Shape& biasShape, in convNhwc() 281 const Shape& filterShape, const _Float16* biasData, const Shape& biasShape, in convNhwc() 308 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in conv() 327 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannelNhwc() 420 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannelNhwc() 484 const Shape& filterShape, const float* filterScales, in convQuant8PerChannel()
|
D | GroupedConv2D.cpp | 47 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/ |
D | OperationsUtils.cpp | 258 bool GetQuantizedConvolutionMultipler(const Shape& inputShape, const Shape& filterShape, in GetQuantizedConvolutionMultipler()
|
D | CpuExecutor.cpp | 1567 Shape filterShape = filter.shape(); in executeOperation() local
|