• Home
  • Raw
  • Download

Lines Matching refs:useNchw

56     bool useNchw = false;  member
69 useNchw = context->getInputValue<bool>(8); in initialize()
86 useNchw = context->getInputValue<bool>(11); in initialize()
98 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
99 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
378 int32_t depthMultiplier, int32_t activation, bool useNchw, T_Input* outputData, in depthwiseConv() argument
380 InputWithLayout<T_Input> input(useNchw); in depthwiseConv()
381 OutputWithLayout<T_Input> output(useNchw); in depthwiseConv()
401 int32_t depthMultiplier, int32_t activation, bool useNchw, in depthwiseConvQuant8PerChannel() argument
403 InputWithLayout<T> input(useNchw); in depthwiseConvQuant8PerChannel()
404 OutputWithLayout<T> output(useNchw); in depthwiseConvQuant8PerChannel()
546 uint32_t height = getSizeOfDimension(input, param.useNchw ? 2 : 1); in prepare()
547 uint32_t width = getSizeOfDimension(input, param.useNchw ? 3 : 2); in prepare()
548 uint32_t channels_in = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare()
570 if (param.useNchw) { in prepare()
594 param.depth_multiplier, param.activation, param.useNchw, in execute()
607 param.depth_multiplier, param.activation, param.useNchw, in execute()
626 param.useNchw, context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
638 param.depth_multiplier, param.activation, param.useNchw, in execute()
660 param.useNchw, context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
673 param.depth_multiplier, param.activation, param.useNchw, in execute()