relu-metal / torch-ext /torch_binding.h
danieldk's picture
danieldk HF Staff
Add sources
f701ae7
#include <torch/torch.h>
torch::Tensor mps_relu(const torch::Tensor &input);