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)); }