Searched refs:lerp_tensor_kernel (Results 1 – 2 of 2) sorted by relevance
13 void lerp_tensor_kernel(at::TensorIteratorBase& iter) { in lerp_tensor_kernel() function124 REGISTER_DISPATCH(lerp_kernel_tensor_weight, &lerp_tensor_kernel);
114 void lerp_tensor_kernel(at::TensorIteratorBase& iter) { in lerp_tensor_kernel() function162 REGISTER_DISPATCH(lerp_kernel_tensor_weight, &lerp_tensor_kernel);