drbh
/

template

Model card Files Files and versions

template / torch-ext /torch_binding.h

drbh

feat: basic template

dda32b8 3 months ago

history blame contribute delete

121 Bytes

	#pragma once

	#include <torch/torch.h>

	void __KERNEL_NAME_NORMALIZED__(torch::Tensor &out, torch::Tensor const &input);