public K210RemovePaddingLayerArgument Convert(K210RemovePadding layer, ConvertContext context)
{
return(new K210RemovePaddingLayerArgument
{
Channels = (uint)layer.Input.Dimensions[1]
});
}
public void Infer(K210RemovePadding layer, K210RemovePaddingLayerArgument argument, InferenceContext context)
{
var inputAlloc = context.MainMemoryMap[layer.Input.Connection.From];
var outputAlloc = context.MainMemoryMap[layer.Output];
argument.Flags = K210LayerFlags.MainMemoryOutput;
argument.MainMemoryInputAddress = inputAlloc.GetAddress();
argument.MainMemoryOutputAddress = outputAlloc.GetAddress();
}
