Searched refs:CpuLogits1DSoftmaxKernel (Results 1 – 6 of 6) sorted by relevance
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[all …]
79 class CpuLogits1DSoftmaxKernel : public ICpuKernel<CpuLogits1DSoftmaxKernel<IS_LOG>>85 CpuLogits1DSoftmaxKernel() = default;86 ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuLogits1DSoftmaxKernel);
96 auto sm = std::make_unique<kernels::CpuLogits1DSoftmaxKernel<IS_LOG>>(); in configure()152 …ARM_COMPUTE_RETURN_ON_ERROR(kernels::CpuLogits1DSoftmaxKernel<IS_LOG>::validate(&tensor_info_tmp, … in validate()
40 class CpuLogits1DSoftmaxKernel; variable
181 …const auto *selected_impl = CpuLogits1DSoftmaxKernel<false>::get_implementation(DataTypeISASelecto…
358 - @ref cpu::kernels::CpuLogits1DSoftmaxKernel