Searched refs:CpuLogits1DMaxKernel (Results 1 – 6 of 6) sorted by relevance
48 static const std::vector<CpuLogits1DMaxKernel::SoftmaxLogits1DMaxKernel> available_kernels_max_logi…108 const std::vector<CpuLogits1DMaxKernel::SoftmaxLogits1DMaxKernel> &CpuLogits1DMaxKernel::get_availa… in get_available_kernels()113 void CpuLogits1DMaxKernel::configure(const ITensorInfo *src, ITensorInfo *dst) in configure()133 Status CpuLogits1DMaxKernel::validate(const ITensorInfo *src, const ITensorInfo *dst) in validate()141 void CpuLogits1DMaxKernel::run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &inf… in run_op()154 const char *CpuLogits1DMaxKernel::name() const in name()
37 class CpuLogits1DMaxKernel : public ICpuKernel<CpuLogits1DMaxKernel>43 CpuLogits1DMaxKernel() = default;44 ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuLogits1DMaxKernel);
92 auto mk = std::make_unique<kernels::CpuLogits1DMaxKernel>(); in configure()151 ARM_COMPUTE_RETURN_ON_ERROR(kernels::CpuLogits1DMaxKernel::validate(src, &tensor_info_max_sum)); in validate()
38 class CpuLogits1DMaxKernel; variable
146 …const auto *selected_impl = CpuLogits1DMaxKernel::get_implementation(DataTypeISASelectorData{ data…
359 - @ref cpu::kernels::CpuLogits1DMaxKernel