/pytorch-coreml-custom-layer-example

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

Primary LanguageSwift