Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/include/
DCpuOperationUtils.h139 class InputWithLayout {
141 InputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in InputWithLayout() function
/frameworks/ml/nn/common/operations/
DPooling.cpp212 InputWithLayout<T> input(param.useNchw); in averagePool()
225 InputWithLayout<T> input(param.useNchw); in l2Pool()
238 InputWithLayout<T> input(param.useNchw); in maxPool()
DInstanceNormalization.cpp80 InputWithLayout<T> input(useNchw); in instanceNorm()
DResizeImageOps.cpp91 InputWithLayout<T> input(useNchw); in resizeImageOp()
DTransposeConv2D.cpp294 InputWithLayout<T_Input> input(param.useNchw); in transposeConv()
412 InputWithLayout<uint8_t> input(param.useNchw); in transposeConvQuant8PerChannel()
DConv2D.cpp269 InputWithLayout<T_Input> input(useNchw); in conv()
383 InputWithLayout<uint8_t> input(useNchw); in convQuant8PerChannel()
DRoiPooling.cpp141 InputWithLayout<T_Input> input(useNchw); in roiPooling()
DRoiAlign.cpp316 InputWithLayout<T_Input> input(useNchw); in roiAlign()
DGenerateProposals.cpp954 InputWithLayout<float> score_nhwc(useNchw), delta_nhwc(useNchw); in generateProposalsFloat32Compute()