Searched refs:unary_pointwise_random_batch_rule (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesRandomness.cpp | 126 Tensor unary_pointwise_random_batch_rule(const Tensor& tensor, ExtraArgs... extra_args) { in unary_pointwise_random_batch_rule() function
325 return unary_pointwise_random_batch_rule<F, Func, T...>(tensor, std::forward<T>(extra_args)...); in apply()
335 return unary_pointwise_random_batch_rule<F, Func, T...>(tensor, std::forward<T>(extra_args)...); in apply()
350 return unary_pointwise_random_batch_rule<decltype(&normal_wrapper<F, Func, T...>), in apply()
|