Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DPooling.cpp259 InputWithLayout<T> input(param.useNchw); in averagePool()
272 InputWithLayout<T> input(param.useNchw); in l2Pool()
285 InputWithLayout<T> input(param.useNchw); in maxPool()
DInstanceNormalization.cpp84 InputWithLayout<T> input(useNchw); in instanceNorm()
DTransposeConv2D.cpp301 InputWithLayout<T_Input> input(param.useNchw); in transposeConv()
421 InputWithLayout<T> input(param.useNchw); in transposeConvQuant8PerChannel()
DResizeImageOps.cpp148 InputWithLayout<T> input(useNchw); in resizeImageOp()
DDepthwiseConv2D.cpp376 InputWithLayout<T_Input> input(useNchw); in depthwiseConv()
399 InputWithLayout<T> input(useNchw); in depthwiseConvQuant8PerChannel()
DRoiPooling.cpp135 InputWithLayout<T_Input> input(useNchw); in roiPooling()
DConv2D.cpp336 InputWithLayout<T_Input> input(useNchw); in conv()
513 InputWithLayout<T> input(useNchw); in convQuant8PerChannel()
DRoiAlign.cpp310 InputWithLayout<T_Input> input(useNchw); in roiAlign()
DGenerateProposals.cpp957 InputWithLayout<float> score_nhwc(useNchw), delta_nhwc(useNchw); in generateProposalsFloat32Compute()
/packages/modules/NeuralNetworks/common/include/
DCpuOperationUtils.h158 class InputWithLayout {
160 InputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in InputWithLayout() function