Home
last modified time | relevance | path

Searched refs:OutputWithLayout (Results 1 – 9 of 9) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DPooling.cpp260 OutputWithLayout<T> output(param.useNchw); in averagePool()
273 OutputWithLayout<T> output(param.useNchw); in l2Pool()
286 OutputWithLayout<T> output(param.useNchw); in maxPool()
DInstanceNormalization.cpp85 OutputWithLayout<T> output(useNchw); in instanceNorm()
DTransposeConv2D.cpp302 OutputWithLayout<T_Input> output(param.useNchw); in transposeConv()
422 OutputWithLayout<T> output(param.useNchw); in transposeConvQuant8PerChannel()
DResizeImageOps.cpp149 OutputWithLayout<T> output(useNchw); in resizeImageOp()
DDepthwiseConv2D.cpp377 OutputWithLayout<T_Input> output(useNchw); in depthwiseConv()
400 OutputWithLayout<T> output(useNchw); in depthwiseConvQuant8PerChannel()
DRoiPooling.cpp136 OutputWithLayout<T_Input> output(useNchw); in roiPooling()
DConv2D.cpp337 OutputWithLayout<T_Input> output(useNchw); in conv()
514 OutputWithLayout<T> output(useNchw); in convQuant8PerChannel()
DRoiAlign.cpp311 OutputWithLayout<T_Input> output(useNchw); in roiAlign()
/packages/modules/NeuralNetworks/common/include/
DCpuOperationUtils.h182 class OutputWithLayout {
184 OutputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in OutputWithLayout() function