Home
last modified time | relevance | path

Searched defs:x_cube (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DActivation.cpp355 float x_cube = float(x) * float(x) * float(x); in GeluKernelImpl() local
384 auto x_cube = x * x * x; in GeluKernelImpl() local
389 auto x_cube = x_vec * x_vec * x_vec; in GeluKernelImpl() local
454 float x_cube = x_sq * float(x); in GeluBackwardKernelImpl() local
517 auto x_cube = x_sq * x; in GeluBackwardKernelImpl() local
535 auto x_cube = x_vec * x_vec * x_vec; in GeluBackwardKernelImpl() local
/aosp_15_r20/external/executorch/kernels/optimized/cpu/
H A Dop_gelu.cpp56 auto x_cube = x * x * x; in gelu() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp775 const auto x_cube = value_dx * value_dx * value_dx; in qgelu_kernel() local