Home
last modified time | relevance | path

Searched refs:histogramdd (Results 1 – 11 of 11) sorted by relevance

/external/pytorch/test/torch_np/numpy_tests/lib/
Dtest_histograms.py24 from numpy import histogram, histogram_bin_edges, histogramdd
35 from torch._numpy import histogram, histogramdd
427 hist = np.histogramdd(sample=sample, bins=(xbins, ybins, zbins))
733 H, edges = histogramdd(x, (2, 3, 3), range=[[-1, 1], [0, 3], [0, 3]])
741 H, edges = histogramdd(x, bins=ed, density=True)
745 H, edges = histogramdd(
758 H, edges = histogramdd(z, bins=(4, 3, 2), range=[[-2, 2], [0, 3], [0, 2]])
771 H, edges = histogramdd([np.arange(5), np.arange(5), np.arange(5)], 5)
779 H, edges = histogramdd(r, b)
813 H, edges = histogramdd(r, b)
[all …]
/external/pytorch/aten/src/ATen/native/mps/operations/
DHistogramKernel.mm55 kernel void histogramdd(constant T * input_ [[buffer(0)]],
127 kernel void histogramdd<DTYPE>( \
175 … TORCH_CHECK(weight.value().is_contiguous(), "histogramdd(): weight should be contiguous on MPS");
232 …TORCH_CHECK(input.numel() * input.element_size() <= UINT32_MAX, "histogramdd(): Tensor is larger t…
351 // histogramdd codepath: both hist and bin_edges are eventually returned as output,
/external/pytorch/aten/src/ATen/native/
DHistogram.cpp416 std::tuple<Tensor, std::vector<Tensor>> histogramdd( in histogramdd() function
424 std::tuple<Tensor, std::vector<Tensor>> histogramdd( in histogramdd() function
433 std::tuple<Tensor, std::vector<Tensor>> histogramdd( in histogramdd() function
437 return at::native::histogramdd(self, bins_v, range, weight, density); in histogramdd()
/external/pytorch/functorch/op_analysis/
Dpublic_api530 histogramdd
/external/pytorch/torch/
D_torch_docs.py5017 torch.histogramdd,
5108 torch.histogramdd.__module__ = "torch"
Doverrides.py676 torch.histogramdd: lambda input, bins, range=None, weight=None, density=False: -1,
/external/pytorch/docs/source/
Dtorch.rst553 histogramdd
/external/pytorch/torch/_numpy/
D_funcs_impl.py1945 h, e = histogramdd((x, y), bins, range, normed, weights, density)
1950 def histogramdd( function
1997 h, b = torch.histogramdd(sample, bins, range, density=bool(density), **w_kwd)
/external/pytorch/test/
Dtest_meta.py676 torch.histogramdd : {f64, f32},
Dtest_reductions.py3226 (np_hist, np_bin_edges) = np.histogramdd(reshaped_t, np_bins,
3231 …(actual_hist, actual_bin_edges) = torch.histogramdd(t, bins, range=bin_range, weight=weights, dens…
/external/pytorch/test/inductor/
Dtest_torchinductor.py8689 return torch.histogramdd(