Searched refs:log_softmax_kernel (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/executorch/kernels/optimized/cpu/ |
| H A D | op_log_softmax.cpp | 32 void log_softmax_kernel(const Tensor& input, int64_t dim, Tensor& out) { in log_softmax_kernel() function
114 log_softmax_kernel<float, OUT_T>(X, dim, out); in log_softmax_wrapper()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | SoftmaxKernel.h | 24 DECLARE_DISPATCH(forward_fn_with_dim, log_softmax_kernel);
|
| H A D | SoftMaxKernel.cpp | 1304 ALSO_REGISTER_AVX512_DISPATCH(log_softmax_kernel, &log_softmax_kernel_impl);
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | SoftMax.cpp | 386 log_softmax_kernel(kCPU, output, input_, dim_); in TORCH_IMPL_FUNC()
560 DEFINE_DISPATCH(log_softmax_kernel);
|