Searched defs:bin_edges (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Histogram.cpp | 127 const Tensor& hist, const TensorList& bin_edges) { in histogramdd_prepare_out()
151 const Tensor& hist, const TensorList& bin_edges) { in histogramdd_prepare_out()
248 Tensor& hist, TensorList& bin_edges) { in histogramdd_out()
308 Tensor& hist, TensorList& bin_edges) { in histogramdd_out()
338 Tensor& hist, Tensor& bin_edges) { in histogram_out()
354 Tensor bin_edges = at::empty({0}, bins.options(), MemoryFormat::Contiguous); in histogram() local
363 Tensor& hist, Tensor& bin_edges) { in histogram_out()
392 Tensor bin_edges = at::empty({0}, self.options()); in histogram_histc_out() local
427 auto bin_edges = at::_histogramdd_bin_edges(self, bins, range, weight, density); in histogramdd() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | HistogramKernel.cpp | 80 void histogramdd_cpu_contiguous(Tensor& hist, const TensorList& bin_edges, in histogramdd_cpu_contiguous()
212 Tensor& hist, const TensorList& bin_edges) { in histogramdd_out_cpu_template()
263 Tensor& hist, const TensorList& bin_edges) { in histogramdd_kernel_impl()
273 bool density, Tensor& hist, const TensorList& bin_edges, bool local_search) { in histogramdd_linear_kernel_impl()
|