dneprDroid / pytorch-coreml-custom-layer-example

PyTorch to CoreML: Writing custom layers with Metal shaders - torch.nn.functional.grid_sample operation

https://medium.com/@ovechko.056/pytorch-to-coreml-writing-custom-layers-with-metal-e8922b51895

Geek Repo

Github PK Tool