Lines Matching refs:useNchw
51 bool useNchw = false; member
67 useNchw = context->getInputValue<bool>(10); in initialize()
77 useNchw = context->getInputValue<bool>(7); in initialize()
82 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
83 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
257 InputWithLayout<T> input(param.useNchw); in averagePool()
258 OutputWithLayout<T> output(param.useNchw); in averagePool()
270 InputWithLayout<T> input(param.useNchw); in l2Pool()
271 OutputWithLayout<T> output(param.useNchw); in l2Pool()
283 InputWithLayout<T> input(param.useNchw); in maxPool()
284 OutputWithLayout<T> output(param.useNchw); in maxPool()
370 uint32_t height = getSizeOfDimension(input, param.useNchw ? 2 : 1); in prepare()
371 uint32_t width = getSizeOfDimension(input, param.useNchw ? 3 : 2); in prepare()
372 uint32_t channels = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare()
383 if (param.useNchw) { in prepare()