Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DPowKernel.cpp53 void pow_tensor_scalar_optimized_kernel(TensorIteratorBase& iter, const exp_scalar_t exp) { in pow_tensor_scalar_optimized_kernel() function
109 pow_tensor_scalar_optimized_kernel<scalar_t, double>( in pow_tensor_scalar_kernel()
114 pow_tensor_scalar_optimized_kernel<scalar_t, scalar_t>( in pow_tensor_scalar_kernel()
132 pow_tensor_scalar_optimized_kernel<scalar_t, scalar_t>( in pow_tensor_scalar_kernel()