Lines Matching refs:useNchw
41 bool useNchw = false; member
57 useNchw = context->getInputValue<bool>(10); in initialize()
67 useNchw = context->getInputValue<bool>(7); in initialize()
72 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
73 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
212 InputWithLayout<T> input(param.useNchw); in averagePool()
213 OutputWithLayout<T> output(param.useNchw); in averagePool()
225 InputWithLayout<T> input(param.useNchw); in l2Pool()
226 OutputWithLayout<T> output(param.useNchw); in l2Pool()
238 InputWithLayout<T> input(param.useNchw); in maxPool()
239 OutputWithLayout<T> output(param.useNchw); in maxPool()
310 uint32_t height = getSizeOfDimension(input, param.useNchw ? 2 : 1); in prepare()
311 uint32_t width = getSizeOfDimension(input, param.useNchw ? 3 : 2); in prepare()
312 uint32_t channels = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare()
323 if (param.useNchw) { in prepare()