Searched refs:_max_shift_exp_sum_kernel (Results 1 – 2 of 2) sorted by relevance
44 _max_shift_exp_sum_kernel(std::make_unique<kernels::ClLogits1DMaxShiftExpSumKernel>()), in ClSoftmax()81 _max_shift_exp_sum_kernel->set_target(CLScheduler::get().target()); in configure()83 …_max_shift_exp_sum_kernel->configure(compile_context, tmp_input_info, _max_info, _tmp_info, _sum_i… in configure()172 CLScheduler::get().enqueue_op(*_max_shift_exp_sum_kernel.get(), sum_pack, false); in run()
80 std::unique_ptr<kernels::ClLogits1DMaxShiftExpSumKernel> _max_shift_exp_sum_kernel; variable