Searched refs:RandIntBatchRuleHelper (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesRandomness.cpp | 281 struct RandIntBatchRuleHelper;
284 struct RandIntBatchRuleHelper<F, Func, typelist<T1, T2, T...>> { struct
390 RandIntBatchRuleHelper<decltype(&ATEN_FN(op)), &ATEN_FN(op), \ in TORCH_LIBRARY_IMPL()
395 RandIntBatchRuleHelper<decltype(&ATEN_FN2(op, overload)), &ATEN_FN2(op, overload), \ in TORCH_LIBRARY_IMPL()
|