public static Padding ToPadding(this tflite.Padding padding)
{
switch (padding)
{
case tflite.Padding.SAME:
return(Padding.Same);
case tflite.Padding.VALID:
return(Padding.Valid);
default:
throw new ArgumentOutOfRangeException(nameof(padding));
}
}
public static Offset <tflite.Conv2DOptions> CreateConv2DOptions(FlatBufferBuilder builder,
tflite.Padding padding = tflite.Padding.SAME,
int stride_w = 0,
int stride_h = 0,
tflite.ActivationFunctionType fused_activation_function = tflite.ActivationFunctionType.NONE,
int dilation_w_factor = 1,
int dilation_h_factor = 1)
{
builder.StartTable(6);
Conv2DOptions.AddDilationHFactor(builder, dilation_h_factor);
Conv2DOptions.AddDilationWFactor(builder, dilation_w_factor);
Conv2DOptions.AddStrideH(builder, stride_h);
Conv2DOptions.AddStrideW(builder, stride_w);
Conv2DOptions.AddFusedActivationFunction(builder, fused_activation_function);
Conv2DOptions.AddPadding(builder, padding);
return(Conv2DOptions.EndConv2DOptions(builder));
}
public static Offset <tflite.Pool2DOptions> CreatePool2DOptions(FlatBufferBuilder builder,
tflite.Padding padding = tflite.Padding.SAME,
int stride_w = 0,
int stride_h = 0,
int filter_width = 0,
int filter_height = 0,
tflite.ActivationFunctionType fused_activation_function = tflite.ActivationFunctionType.NONE)
{
builder.StartTable(6);
Pool2DOptions.AddFilterHeight(builder, filter_height);
Pool2DOptions.AddFilterWidth(builder, filter_width);
Pool2DOptions.AddStrideH(builder, stride_h);
Pool2DOptions.AddStrideW(builder, stride_w);
Pool2DOptions.AddFusedActivationFunction(builder, fused_activation_function);
Pool2DOptions.AddPadding(builder, padding);
return(Pool2DOptions.EndPool2DOptions(builder));
}
public static void AddPadding(FlatBufferBuilder builder, tflite.Padding padding)
{
builder.AddSbyte(0, (sbyte)padding, 0);
}
public static Offset <tflite.TransposeConvOptions> CreateTransposeConvOptions(FlatBufferBuilder builder,
tflite.Padding padding = tflite.Padding.SAME,
int stride_w = 0,
int stride_h = 0)
{
builder.StartTable(3);
TransposeConvOptions.AddStrideH(builder, stride_h);
TransposeConvOptions.AddStrideW(builder, stride_w);
TransposeConvOptions.AddPadding(builder, padding);
return(TransposeConvOptions.EndTransposeConvOptions(builder));
}
