Searched refs:RandomInplaceBatchRuleHelper (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesRandomness.cpp | 271 struct RandomInplaceBatchRuleHelper;
274 struct RandomInplaceBatchRuleHelper<F, Func, typelist<T1, T...>> { struct
359 RandomInplaceBatchRuleHelper<decltype(&ATEN_FN2(op, overload)), &ATEN_FN2(op, overload), \ in TORCH_LIBRARY_IMPL()
380 RandomInplaceBatchRuleHelper<decltype(&ATEN_FN(op)), &ATEN_FN(op), \ in TORCH_LIBRARY_IMPL()
385 RandomInplaceBatchRuleHelper<decltype(&ATEN_FN2(op, overload)), &ATEN_FN2(op, overload), \ in TORCH_LIBRARY_IMPL()
|