Exemplo n.º 1
0
 public K210RemovePaddingLayerArgument Convert(K210RemovePadding layer, ConvertContext context)
 {
     return(new K210RemovePaddingLayerArgument
     {
         Channels = (uint)layer.Input.Dimensions[1]
     });
 }
Exemplo n.º 2
0
        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();
        }