Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddebug_events_monitors.py90 num_nan=None, argument
128 def num_nan(self): member in InfNanAlert
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DSorting.cpp587 int64_t num_nan = std::count_if(ip, ip + size, _isnan<scalar_t>); in median_with_indices_impl() local
638 int64_t num_nan = std::count_if(first, last, _isnan<scalar_t>); in median_impl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSorting.cu129 __shared__ int64_t num_nan; in gatherMedian() local