Lines Matching refs:useNchw
54 bool useNchw = false; member
64 useNchw = context->getInputValue<bool>(8); in initialize()
71 int32_t outputWidth = useNchw ? outputShapeData[3] : outputShapeData[2]; in initialize()
72 int32_t outputHeight = useNchw ? outputShapeData[2] : outputShapeData[1]; in initialize()
85 useNchw = context->getInputValue<bool>(10); in initialize()
294 InputWithLayout<T_Input> input(param.useNchw); in transposeConv()
295 OutputWithLayout<T_Input> output(param.useNchw); in transposeConv()
412 InputWithLayout<uint8_t> input(param.useNchw); in transposeConvQuant8PerChannel()
413 OutputWithLayout<uint8_t> output(param.useNchw); in transposeConvQuant8PerChannel()
488 uint32_t height = getSizeOfDimension(input, param.useNchw ? 2 : 1); in prepare()
489 uint32_t width = getSizeOfDimension(input, param.useNchw ? 3 : 2); in prepare()
490 uint32_t channels_in = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare()
513 if (param.useNchw) { in prepare()