replicate/megablocks
0122
1#pragma once2 3#include <torch/all.h>4 5namespace megablocks {6 7// Public interface function for computing histograms8torch::Tensor histogram(torch::Tensor x, int num_bins);9 10} // namespace megablocks1#pragma once2 3#include <torch/all.h>4 5namespace megablocks {6 7// Public interface function for computing histograms8torch::Tensor histogram(torch::Tensor x, int num_bins);9 10} // namespace megablocks