Home
last modified time | relevance | path

Searched refs:renorm_scale_factor_impl (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DRenormKernel.cu11 void renorm_scale_factor_impl(TensorIteratorBase& iter, double maxnorm) { in renorm_scale_factor_impl() function
27 REGISTER_DISPATCH(renorm_scale_factor_stub, &renorm_scale_factor_impl);
/external/pytorch/aten/src/ATen/native/cpu/
DRenormKernel.cpp13 void renorm_scale_factor_impl(TensorIteratorBase& iter, double maxnorm) { in renorm_scale_factor_impl() function
36 REGISTER_DISPATCH(renorm_scale_factor_stub, &renorm_scale_factor_impl);