Home
last modified time | relevance | path

Searched refs:InputWithLayout (Results 1 – 10 of 10) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/include/
H A DCpuOperationUtils.h158 class InputWithLayout {
160 InputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in InputWithLayout() function
/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DPooling.cpp257 InputWithLayout<T> input(param.useNchw); in averagePool()
270 InputWithLayout<T> input(param.useNchw); in l2Pool()
283 InputWithLayout<T> input(param.useNchw); in maxPool()
H A DInstanceNormalization.cpp94 InputWithLayout<T> input(useNchw); in instanceNorm()
H A DTransposeConv2D.cpp305 InputWithLayout<T_Input> input(param.useNchw); in transposeConv()
425 InputWithLayout<T> input(param.useNchw); in transposeConvQuant8PerChannel()
H A DDepthwiseConv2D.cpp380 InputWithLayout<T_Input> input(useNchw); in depthwiseConv()
403 InputWithLayout<T> input(useNchw); in depthwiseConvQuant8PerChannel()
H A DResizeImageOps.cpp155 InputWithLayout<T> input(useNchw); in resizeImageOp()
H A DRoiPooling.cpp148 InputWithLayout<T_Input> input(useNchw); in roiPooling()
H A DConv2D.cpp339 InputWithLayout<T_Input> input(useNchw); in conv()
516 InputWithLayout<T> input(useNchw); in convQuant8PerChannel()
H A DRoiAlign.cpp322 InputWithLayout<T_Input> input(useNchw); in roiAlign()
H A DGenerateProposals.cpp1070 InputWithLayout<float> score_nhwc(useNchw), delta_nhwc(useNchw); in generateProposalsFloat32Compute()