Searched refs:DiscreteDistributionType (Results 1 – 2 of 2) sorted by relevance
229 struct DiscreteDistributionType { using type = float; }; argument231 template <> struct DiscreteDistributionType<double> { using type = double; };
516 using accscalar_t = at::DiscreteDistributionType<scalar_t>::type; in geometric_kernel()652 using accscalar_t = at::DiscreteDistributionType<scalar_t>::type; in bernoulli_kernel()