Lines Matching refs:useNchw

67     bool useNchw = false;  member
79 useNchw = context->getInputValue<bool>(7); in initialize()
95 useNchw = context->getInputValue<bool>(10); in initialize()
107 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
108 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
334 int32_t dilation_height_factor, int32_t activation, bool useNchw, T_Input* outputData, in conv() argument
336 InputWithLayout<T_Input> input(useNchw); in conv()
337 OutputWithLayout<T_Input> output(useNchw); in conv()
511 int32_t dilationHeightFactor, int32_t activation, bool useNchw, in convQuant8PerChannel() argument
513 InputWithLayout<T> input(useNchw); in convQuant8PerChannel()
514 OutputWithLayout<T> output(useNchw); in convQuant8PerChannel()
555 uint32_t height = getSizeOfDimension(input, param.useNchw ? 2 : 1); in prepare()
556 uint32_t width = getSizeOfDimension(input, param.useNchw ? 3 : 2); in prepare()
557 uint32_t channels_in = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare()
585 if (param.useNchw) { in prepare()
608 param.dilation_height_factor, param.activation, param.useNchw, in execute()
620 param.dilation_height_factor, param.activation, param.useNchw, in execute()
638 param.dilation_height_factor, param.activation, param.useNchw, in execute()
650 param.dilation_height_factor, param.activation, param.useNchw, in execute()
671 param.dilation_height_factor, param.activation, param.useNchw, in execute()
684 param.dilation_height_factor, param.activation, param.useNchw, in execute()