Lines Matching refs:useNchw
53 bool useNchw = false; member
69 useNchw = context->getInputValue<bool>(10); in initialize()
79 useNchw = context->getInputValue<bool>(7); in initialize()
84 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
85 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
259 InputWithLayout<T> input(param.useNchw); in averagePool()
260 OutputWithLayout<T> output(param.useNchw); in averagePool()
272 InputWithLayout<T> input(param.useNchw); in l2Pool()
273 OutputWithLayout<T> output(param.useNchw); in l2Pool()
285 InputWithLayout<T> input(param.useNchw); in maxPool()
286 OutputWithLayout<T> output(param.useNchw); in maxPool()
306 uint32_t height = getSizeOfDimension(input, param.useNchw ? 2 : 1); in prepare()
307 uint32_t width = getSizeOfDimension(input, param.useNchw ? 3 : 2); in prepare()
308 uint32_t channels = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare()
319 if (param.useNchw) { in prepare()