public static Offset <tflite.DivOptions> CreateDivOptions(FlatBufferBuilder builder,
tflite.ActivationFunctionType fused_activation_function = tflite.ActivationFunctionType.NONE)
{
builder.StartTable(1);
DivOptions.AddFusedActivationFunction(builder, fused_activation_function);
return(DivOptions.EndDivOptions(builder));
}
public static Offset <tflite.SVDFOptions> CreateSVDFOptions(FlatBufferBuilder builder,
int rank = 0,
tflite.ActivationFunctionType fused_activation_function = tflite.ActivationFunctionType.NONE)
{
builder.StartTable(2);
SVDFOptions.AddRank(builder, rank);
SVDFOptions.AddFusedActivationFunction(builder, fused_activation_function);
return(SVDFOptions.EndSVDFOptions(builder));
}
public static Offset <tflite.SequenceRNNOptions> CreateSequenceRNNOptions(FlatBufferBuilder builder,
bool time_major = false,
tflite.ActivationFunctionType fused_activation_function = tflite.ActivationFunctionType.NONE)
{
builder.StartTable(2);
SequenceRNNOptions.AddFusedActivationFunction(builder, fused_activation_function);
SequenceRNNOptions.AddTimeMajor(builder, time_major);
return(SequenceRNNOptions.EndSequenceRNNOptions(builder));
}
public static Offset <tflite.LSTMOptions> CreateLSTMOptions(FlatBufferBuilder builder,
tflite.ActivationFunctionType fused_activation_function = tflite.ActivationFunctionType.NONE,
float cell_clip = 0.0f,
float proj_clip = 0.0f,
tflite.LSTMKernelType kernel_type = tflite.LSTMKernelType.FULL)
{
builder.StartTable(4);
LSTMOptions.AddProjClip(builder, proj_clip);
LSTMOptions.AddCellClip(builder, cell_clip);
LSTMOptions.AddKernelType(builder, kernel_type);
LSTMOptions.AddFusedActivationFunction(builder, fused_activation_function);
return(LSTMOptions.EndLSTMOptions(builder));
}
public static ActivationFunctionType ToActivationFunction(this tflite.ActivationFunctionType activation)
{
switch (activation)
{
case tflite.ActivationFunctionType.NONE:
return(ActivationFunctionType.Linear);
case tflite.ActivationFunctionType.RELU:
return(ActivationFunctionType.Relu);
case tflite.ActivationFunctionType.RELU6:
return(ActivationFunctionType.Relu6);
default:
throw new NotSupportedException(nameof(activation));
}
}
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 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 void AddFusedActivationFunction(FlatBufferBuilder builder, tflite.ActivationFunctionType fusedActivationFunction)
{
builder.AddSbyte(0, (sbyte)fusedActivationFunction, 0);
}
public static Offset <tflite.FullyConnectedOptions> CreateFullyConnectedOptions(FlatBufferBuilder builder,
tflite.ActivationFunctionType fused_activation_function = tflite.ActivationFunctionType.NONE,
tflite.FullyConnectedOptionsWeightsFormat weights_format = tflite.FullyConnectedOptionsWeightsFormat.DEFAULT,
bool keep_num_dims = false)
{
builder.StartTable(3);
FullyConnectedOptions.AddKeepNumDims(builder, keep_num_dims);
FullyConnectedOptions.AddWeightsFormat(builder, weights_format);
FullyConnectedOptions.AddFusedActivationFunction(builder, fused_activation_function);
return(FullyConnectedOptions.EndFullyConnectedOptions(builder));
}
public static Offset <tflite.BidirectionalSequenceLSTMOptions> CreateBidirectionalSequenceLSTMOptions(FlatBufferBuilder builder,
tflite.ActivationFunctionType fused_activation_function = tflite.ActivationFunctionType.NONE,
float cell_clip = 0.0f,
float proj_clip = 0.0f,
bool merge_outputs = false,
bool time_major = true)
{
builder.StartTable(5);
BidirectionalSequenceLSTMOptions.AddProjClip(builder, proj_clip);
BidirectionalSequenceLSTMOptions.AddCellClip(builder, cell_clip);
BidirectionalSequenceLSTMOptions.AddTimeMajor(builder, time_major);
BidirectionalSequenceLSTMOptions.AddMergeOutputs(builder, merge_outputs);
BidirectionalSequenceLSTMOptions.AddFusedActivationFunction(builder, fused_activation_function);
return(BidirectionalSequenceLSTMOptions.EndBidirectionalSequenceLSTMOptions(builder));
}
public static Offset <tflite.UnidirectionalSequenceLSTMOptions> CreateUnidirectionalSequenceLSTMOptions(FlatBufferBuilder builder,
tflite.ActivationFunctionType fused_activation_function = tflite.ActivationFunctionType.NONE,
float cell_clip = 0.0f,
float proj_clip = 0.0f,
bool time_major = false)
{
builder.StartTable(4);
UnidirectionalSequenceLSTMOptions.AddProjClip(builder, proj_clip);
UnidirectionalSequenceLSTMOptions.AddCellClip(builder, cell_clip);
UnidirectionalSequenceLSTMOptions.AddTimeMajor(builder, time_major);
UnidirectionalSequenceLSTMOptions.AddFusedActivationFunction(builder, fused_activation_function);
return(UnidirectionalSequenceLSTMOptions.EndUnidirectionalSequenceLSTMOptions(builder));
}
public static Offset <tflite.ConcatenationOptions> CreateConcatenationOptions(FlatBufferBuilder builder,
int axis = 0,
tflite.ActivationFunctionType fused_activation_function = tflite.ActivationFunctionType.NONE)
{
builder.StartTable(2);
ConcatenationOptions.AddAxis(builder, axis);
ConcatenationOptions.AddFusedActivationFunction(builder, fused_activation_function);
return(ConcatenationOptions.EndConcatenationOptions(builder));
}
public static Offset <tflite.BidirectionalSequenceRNNOptions> CreateBidirectionalSequenceRNNOptions(FlatBufferBuilder builder,
bool time_major = false,
tflite.ActivationFunctionType fused_activation_function = tflite.ActivationFunctionType.NONE,
bool merge_outputs = false)
{
builder.StartTable(3);
BidirectionalSequenceRNNOptions.AddMergeOutputs(builder, merge_outputs);
BidirectionalSequenceRNNOptions.AddFusedActivationFunction(builder, fused_activation_function);
BidirectionalSequenceRNNOptions.AddTimeMajor(builder, time_major);
return(BidirectionalSequenceRNNOptions.EndBidirectionalSequenceRNNOptions(builder));
}
