Lines Matching refs:useNchw
59 bool useNchw = false; member
69 useNchw = context->getInputValue<bool>(8); in initialize()
76 int32_t outputWidth = useNchw ? outputShapeData[3] : outputShapeData[2]; in initialize()
77 int32_t outputHeight = useNchw ? outputShapeData[2] : outputShapeData[1]; in initialize()
90 useNchw = context->getInputValue<bool>(10); in initialize()
301 InputWithLayout<T_Input> input(param.useNchw); in transposeConv()
302 OutputWithLayout<T_Input> output(param.useNchw); in transposeConv()
421 InputWithLayout<T> input(param.useNchw); in transposeConvQuant8PerChannel()
422 OutputWithLayout<T> output(param.useNchw); in transposeConvQuant8PerChannel()
461 uint32_t height = getSizeOfDimension(input, param.useNchw ? 2 : 1); in prepare()
462 uint32_t width = getSizeOfDimension(input, param.useNchw ? 3 : 2); in prepare()
463 uint32_t channels_in = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare()
486 if (param.useNchw) { in prepare()