Lines Matching refs:useNchw
52 bool useNchw = false; member
65 useNchw = context->getInputValue<bool>(8); in initialize()
82 useNchw = context->getInputValue<bool>(11); in initialize()
94 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
95 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
374 int32_t depthMultiplier, int32_t activation, bool useNchw, T_Input* outputData, in depthwiseConv() argument
376 InputWithLayout<T_Input> input(useNchw); in depthwiseConv()
377 OutputWithLayout<T_Input> output(useNchw); in depthwiseConv()
397 int32_t depthMultiplier, int32_t activation, bool useNchw, in depthwiseConvQuant8PerChannel() argument
399 InputWithLayout<T> input(useNchw); in depthwiseConvQuant8PerChannel()
400 OutputWithLayout<T> output(useNchw); in depthwiseConvQuant8PerChannel()
443 uint32_t height = getSizeOfDimension(input, param.useNchw ? 2 : 1); in prepare()
444 uint32_t width = getSizeOfDimension(input, param.useNchw ? 3 : 2); in prepare()
445 uint32_t channels_in = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare()
467 if (param.useNchw) { in prepare()
491 param.depth_multiplier, param.activation, param.useNchw, in execute()
504 param.depth_multiplier, param.activation, param.useNchw, in execute()
523 param.useNchw, context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
535 param.depth_multiplier, param.activation, param.useNchw, in execute()
557 param.useNchw, context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
570 param.depth_multiplier, param.activation, param.useNchw, in execute()