Kernels:

kernels-community
/

megablocks

Trusted publisher

megablocks / csrc /new_histogram.h

drbh

Migrated from kernels-community/megablocks

29caf74 unverified 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