Home
last modified time | relevance | path

Searched defs:ignore_nan (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DSorting.cu99 bool ignore_nan) { in gatherMedian()
209 bool ignore_nan; member
265 const TensorBase &self, int64_t dim, bool ignore_nan) { in launch_median_kernel()
DSorting.cpp81 bool ignore_nan) { in median_with_indices_impl()
133 Tensor median_impl(const Tensor& self, bool ignore_nan) { in median_impl()
/external/pytorch/aten/src/ATen/native/
DSorting.cpp252 const bool ignore_nan, in quantile_compute()
378 const bool ignore_nan) { in quantile_out_impl()
403 const bool ignore_nan) { in quantile_impl()
507 bool ignore_nan) { in median_with_indices_impl()
607 Tensor median_impl(const Tensor& self, bool ignore_nan) { in median_impl()