Lines Matching refs:CpuLogits1DSoftmaxKernel
161 static const std::vector<typename CpuLogits1DSoftmaxKernel<IS_LOG>::SoftmaxLogits1DKernel> availabl…
246 const std::vector<typename CpuLogits1DSoftmaxKernel<IS_LOG>::SoftmaxLogits1DKernel> &CpuLogits1DSof… in get_available_kernels()
252 void CpuLogits1DSoftmaxKernel<IS_LOG>::configure(const ITensorInfo *src, const ITensorInfo *max, IT… in configure()
268 …const auto *uk = CpuLogits1DSoftmaxKernel<IS_LOG>::get_implementation(DataTypeISASelectorData{ src… in configure()
280 ICpuKernel<CpuLogits1DSoftmaxKernel<IS_LOG>>::configure(win); in configure()
284 Status CpuLogits1DSoftmaxKernel<IS_LOG>::validate(const ITensorInfo *src, const ITensorInfo *max, in validate()
294 void CpuLogits1DSoftmaxKernel<IS_LOG>::run_op(ITensorPack &tensors, const Window &window, const Thr… in run_op()
298 …ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICpuKernel<CpuLogits1DSoftmaxKernel<IS_LOG>>::window(), win… in run_op()
316 const char *CpuLogits1DSoftmaxKernel<IS_LOG>::name() const in name()
321 template class CpuLogits1DSoftmaxKernel<true>; variable
322 template class CpuLogits1DSoftmaxKernel<false>; variable