Searched refs:OutputWithLayout (Results 1 – 9 of 9) sorted by relevance
260 OutputWithLayout<T> output(param.useNchw); in averagePool()273 OutputWithLayout<T> output(param.useNchw); in l2Pool()286 OutputWithLayout<T> output(param.useNchw); in maxPool()
85 OutputWithLayout<T> output(useNchw); in instanceNorm()
302 OutputWithLayout<T_Input> output(param.useNchw); in transposeConv()422 OutputWithLayout<T> output(param.useNchw); in transposeConvQuant8PerChannel()
149 OutputWithLayout<T> output(useNchw); in resizeImageOp()
377 OutputWithLayout<T_Input> output(useNchw); in depthwiseConv()400 OutputWithLayout<T> output(useNchw); in depthwiseConvQuant8PerChannel()
136 OutputWithLayout<T_Input> output(useNchw); in roiPooling()
337 OutputWithLayout<T_Input> output(useNchw); in conv()514 OutputWithLayout<T> output(useNchw); in convQuant8PerChannel()
311 OutputWithLayout<T_Input> output(useNchw); in roiAlign()
182 class OutputWithLayout {184 OutputWithLayout(bool useNchw) : mDataOriginal(nullptr), mUseNchw(useNchw) {} in OutputWithLayout() function