ahadnagy
/

megablocks

Model card Files Files and versions

megablocks / csrc /new_histogram.h

Akos Hadnagy

WIP

1e1ffe8 about 2 months ago

198 Bytes

	#pragma once

	#include <torch/all.h>

	namespace megablocks {

	// Public interface function for computing histograms
	torch::Tensor histogram(torch::Tensor x, int num_bins);

	} // namespace megablocks