Searched refs:_max_shift_exp_sum_kernel (Results 1 – 2 of 2) sorted by relevance
45 _max_shift_exp_sum_kernel(support::cpp14::make_unique<CLLogits1DMaxShiftExpSumKernel>()), in CLSoftmaxLayerGeneric()95 _max_shift_exp_sum_kernel->set_target(CLScheduler::get().target()); in configure()108 …_max_shift_exp_sum_kernel->configure(compile_context, tmp_input, &_max, &_tmp, &_sum, softmax_info… in configure()173 CLScheduler::get().enqueue(*_max_shift_exp_sum_kernel, false); in run()
111 std::unique_ptr<CLLogits1DMaxShiftExpSumKernel> _max_shift_exp_sum_kernel; variable