Searched refs:pow_tensor_scalar_kernel (Results 1 – 2 of 2) sorted by relevance
21 void pow_tensor_scalar_kernel(TensorIteratorBase& iter, const Scalar& exp_scalar);134 pow_tensor_scalar_kernel(iter, exp); in pow_tensor_tensor_kernel()170 void pow_tensor_scalar_kernel(TensorIteratorBase& iter, const Scalar& exp_scalar) { in pow_tensor_scalar_kernel() function207 REGISTER_DISPATCH(pow_tensor_scalar_stub, &pow_tensor_scalar_kernel);
89 static void pow_tensor_scalar_kernel( in pow_tensor_scalar_kernel() function148 ALSO_REGISTER_AVX512_DISPATCH(pow_tensor_scalar_stub, &CPU_CAPABILITY::pow_tensor_scalar_kernel);