Lines Matching refs:param
534 DepthwiseConv2dParam param; in prepare() local
535 NN_RET_CHECK(param.initialize(context)); in prepare()
538 uint32_t height = getSizeOfDimension(input, param.useNchw ? 2 : 1); in prepare()
539 uint32_t width = getSizeOfDimension(input, param.useNchw ? 3 : 2); in prepare()
540 uint32_t channels_in = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare()
545 NN_OPS_CHECK(param.depth_multiplier * channels_in == channels_out); in prepare()
546 int32_t effectiveFilterWidth = (filterWidth - 1) * param.dilation_width_factor + 1; in prepare()
547 int32_t effectiveFilterHeight = (filterHeight - 1) * param.dilation_height_factor + 1; in prepare()
548 NN_RET_CHECK_GT(effectiveFilterWidth, param.padding_left); in prepare()
549 NN_RET_CHECK_GT(effectiveFilterWidth, param.padding_right); in prepare()
550 NN_RET_CHECK_GT(effectiveFilterHeight, param.padding_top); in prepare()
551 NN_RET_CHECK_GT(effectiveFilterHeight, param.padding_bottom); in prepare()
554 computeOutSize(height, filterHeight, param.stride_height, param.dilation_height_factor, in prepare()
555 param.padding_top, param.padding_bottom); in prepare()
557 computeOutSize(width, filterWidth, param.stride_width, param.dilation_width_factor, in prepare()
558 param.padding_left, param.padding_right); in prepare()
562 if (param.useNchw) { in prepare()
573 DepthwiseConv2dParam param; in execute() local
574 NN_RET_CHECK(param.initialize(context)); in execute()
582 context->getInputShape(kBiasTensor), param.padding_left, in execute()
583 param.padding_right, param.padding_top, param.padding_bottom, in execute()
584 param.stride_width, param.stride_height, in execute()
585 param.dilation_width_factor, param.dilation_height_factor, in execute()
586 param.depth_multiplier, param.activation, param.useNchw, in execute()
595 context->getInputShape(kBiasTensor), param.padding_left, in execute()
596 param.padding_right, param.padding_top, param.padding_bottom, in execute()
597 param.stride_width, param.stride_height, in execute()
598 param.dilation_width_factor, param.dilation_height_factor, in execute()
599 param.depth_multiplier, param.activation, param.useNchw, in execute()
612 context->getInputShape(kBiasTensor), param.padding_left, in execute()
613 param.padding_right, param.padding_top, param.padding_bottom, in execute()
614 param.stride_width, param.stride_height, param.dilation_width_factor, in execute()
615 param.dilation_height_factor, param.depth_multiplier, param.activation, in execute()
616 param.useNchw, context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
624 context->getInputShape(kBiasTensor), param.padding_left, in execute()
625 param.padding_right, param.padding_top, param.padding_bottom, in execute()
626 param.stride_width, param.stride_height, in execute()
627 param.dilation_width_factor, param.dilation_height_factor, in execute()
628 param.depth_multiplier, param.activation, param.useNchw, in execute()
644 context->getInputShape(kBiasTensor), param.padding_left, in execute()
645 param.padding_right, param.padding_top, param.padding_bottom, in execute()
646 param.stride_width, param.stride_height, param.dilation_width_factor, in execute()
647 param.dilation_height_factor, param.depth_multiplier, param.activation, in execute()
648 param.useNchw, context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
657 context->getInputShape(kBiasTensor), param.padding_left, in execute()
658 param.padding_right, param.padding_top, param.padding_bottom, in execute()
659 param.stride_width, param.stride_height, in execute()
660 param.dilation_width_factor, param.dilation_height_factor, in execute()
661 param.depth_multiplier, param.activation, param.useNchw, in execute()