Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/include/
DCpuOperationUtils.h163 class OutputWithLayout {
165 OutputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in OutputWithLayout() function
/frameworks/ml/nn/common/operations/
DPooling.cpp213 OutputWithLayout<T> output(param.useNchw); in averagePool()
226 OutputWithLayout<T> output(param.useNchw); in l2Pool()
239 OutputWithLayout<T> output(param.useNchw); in maxPool()
DInstanceNormalization.cpp81 OutputWithLayout<T> output(useNchw); in instanceNorm()
DResizeImageOps.cpp92 OutputWithLayout<T> output(useNchw); in resizeImageOp()
DTransposeConv2D.cpp295 OutputWithLayout<T_Input> output(param.useNchw); in transposeConv()
413 OutputWithLayout<uint8_t> output(param.useNchw); in transposeConvQuant8PerChannel()
DConv2D.cpp270 OutputWithLayout<T_Input> output(useNchw); in conv()
384 OutputWithLayout<uint8_t> output(useNchw); in convQuant8PerChannel()
DRoiPooling.cpp142 OutputWithLayout<T_Input> output(useNchw); in roiPooling()
DRoiAlign.cpp317 OutputWithLayout<T_Input> output(useNchw); in roiAlign()