Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/include/
H A DCpuOperationUtils.h182 class OutputWithLayout {
184 OutputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in OutputWithLayout() function
/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DPooling.cpp258 OutputWithLayout<T> output(param.useNchw); in averagePool()
271 OutputWithLayout<T> output(param.useNchw); in l2Pool()
284 OutputWithLayout<T> output(param.useNchw); in maxPool()
H A DInstanceNormalization.cpp95 OutputWithLayout<T> output(useNchw); in instanceNorm()
H A DTransposeConv2D.cpp306 OutputWithLayout<T_Input> output(param.useNchw); in transposeConv()
426 OutputWithLayout<T> output(param.useNchw); in transposeConvQuant8PerChannel()
H A DDepthwiseConv2D.cpp381 OutputWithLayout<T_Input> output(useNchw); in depthwiseConv()
404 OutputWithLayout<T> output(useNchw); in depthwiseConvQuant8PerChannel()
H A DResizeImageOps.cpp156 OutputWithLayout<T> output(useNchw); in resizeImageOp()
H A DRoiPooling.cpp149 OutputWithLayout<T_Input> output(useNchw); in roiPooling()
H A DConv2D.cpp340 OutputWithLayout<T_Input> output(useNchw); in conv()
517 OutputWithLayout<T> output(useNchw); in convQuant8PerChannel()
H A DRoiAlign.cpp323 OutputWithLayout<T_Input> output(useNchw); in roiAlign()