Minor refactoring.
Made handling of optional dilation parameters in 2D convolution identical to depthwise 2D convolution's. PiperOrigin-RevId: 288040518 Change-Id: I343f000917e617cdf416ea6b94e842087b368f9c
This commit is contained in:
parent
9b32b2db11
commit
05dd398ea5
@ -2185,27 +2185,18 @@ TfLiteStatus NNAPIDelegateKernel::Map(
|
||||
case kTfLiteBuiltinConv2d: {
|
||||
auto builtin =
|
||||
reinterpret_cast<TfLiteConvParams*>(mapping_args.node->builtin_data);
|
||||
mapping_args.builder->AddScalarInt32Operand(builtin->padding);
|
||||
mapping_args.builder->AddScalarInt32Operand(builtin->stride_width);
|
||||
mapping_args.builder->AddScalarInt32Operand(builtin->stride_height);
|
||||
mapping_args.builder->AddScalarInt32Operand(builtin->activation);
|
||||
// NNAPI supports dilated Conv2D since NNAPI 1.2.
|
||||
if (builtin->dilation_width_factor != 1 ||
|
||||
builtin->dilation_height_factor != 1) {
|
||||
auto builtin = reinterpret_cast<TfLiteConvParams*>(
|
||||
mapping_args.node->builtin_data);
|
||||
mapping_args.builder->AddScalarInt32Operand(builtin->padding);
|
||||
mapping_args.builder->AddScalarInt32Operand(builtin->stride_width);
|
||||
mapping_args.builder->AddScalarInt32Operand(builtin->stride_height);
|
||||
mapping_args.builder->AddScalarInt32Operand(builtin->activation);
|
||||
mapping_args.builder->AddScalarBoolOperand(false); // Use NHWC format
|
||||
mapping_args.builder->AddScalarInt32Operand(
|
||||
builtin->dilation_width_factor);
|
||||
mapping_args.builder->AddScalarInt32Operand(
|
||||
builtin->dilation_height_factor);
|
||||
} else {
|
||||
auto builtin = reinterpret_cast<TfLiteConvParams*>(
|
||||
mapping_args.node->builtin_data);
|
||||
mapping_args.builder->AddScalarInt32Operand(builtin->padding);
|
||||
mapping_args.builder->AddScalarInt32Operand(builtin->stride_width);
|
||||
mapping_args.builder->AddScalarInt32Operand(builtin->stride_height);
|
||||
mapping_args.builder->AddScalarInt32Operand(builtin->activation);
|
||||
}
|
||||
*nn_op_type = ANEURALNETWORKS_CONV_2D;
|
||||
} break;
|
||||
|
Loading…
x
Reference in New Issue
Block a user