Home
last modified time | relevance | path

Searched refs:filterDepth (Results 1 – 3 of 3) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DGroupedConv2D.cpp44 uint32_t filterDepth = getSizeOfDimension(filterShape, 3); \
77 for (uint32_t k = 0; k < filterDepth; k++) { in groupedConvFloat32()
80 uint32_t dInput = filterDepth * g + k; in groupedConvFloat32()
84 i * filterWidth * filterDepth + j * filterDepth + k; in groupedConvFloat32()
95 filterBase += filterHeight * filterWidth * filterDepth; in groupedConvFloat32()
149 for (uint32_t k = 0; k < filterDepth; k++) { in groupedConvQuant8()
152 uint32_t dInput = filterDepth * g + k; in groupedConvQuant8()
156 i * filterWidth * filterDepth + j * filterDepth + k; in groupedConvQuant8()
173 filterBase += filterHeight * filterWidth * filterDepth; in groupedConvQuant8()
254 for (uint32_t k = 0; k < filterDepth; k++) { in groupedConvQuant8PerChannel()
[all …]
DConv2D.cpp365 uint32_t filterDepth = getSizeOfDimension(filterShape, 3); in convQuant8PerChannelNhwc() local
406 for (uint32_t k = 0; k < filterDepth; k++) { in convQuant8PerChannelNhwc()
415 i * filterWidth * filterDepth + j * filterDepth + k; in convQuant8PerChannelNhwc()
431 filterBase += filterHeight * filterWidth * filterDepth; in convQuant8PerChannelNhwc()
458 [[maybe_unused]] uint32_t filterDepth = getSizeOfDimension(filterShape, 3); in convQuant8PerChannelNhwc() local
DDepthwiseConv2D.cpp291 uint32_t filterDepth = getSizeOfDimension(filterShape, 3); in depthwiseConvQuant8PerChannelNhwc() local
341 i * filterWidth * filterDepth + j * filterDepth + oc; in depthwiseConvQuant8PerChannelNhwc()