Searched refs:CLLogits1DMaxShiftExpSumKernel (Results 1 – 5 of 5) sorted by relevance
35 class CLLogits1DMaxShiftExpSumKernel : public ICLKernel43 CLLogits1DMaxShiftExpSumKernel();45 CLLogits1DMaxShiftExpSumKernel(const CLLogits1DMaxShiftExpSumKernel &) = delete;47 CLLogits1DMaxShiftExpSumKernel &operator=(const CLLogits1DMaxShiftExpSumKernel &) = delete;49 CLLogits1DMaxShiftExpSumKernel(CLLogits1DMaxShiftExpSumKernel &&) = default;51 CLLogits1DMaxShiftExpSumKernel &operator=(CLLogits1DMaxShiftExpSumKernel &&) = default;
147 const unsigned int CLLogits1DMaxShiftExpSumKernel::_grid_size = 64;149 const unsigned int CLLogits1DMaxShiftExpSumKernel::_serial_vector_size = 8;151 const unsigned int CLLogits1DMaxShiftExpSumKernel::_parallel_vector_size = 4;153 CLLogits1DMaxShiftExpSumKernel::CLLogits1DMaxShiftExpSumKernel() in CLLogits1DMaxShiftExpSumKernel() function in arm_compute::CLLogits1DMaxShiftExpSumKernel158 void CLLogits1DMaxShiftExpSumKernel::configure(const ICLTensor *input, ICLTensor *max, ICLTensor *o… in configure()163 void CLLogits1DMaxShiftExpSumKernel::configure(const CLCompileContext &compile_context, const ICLTe… in configure()238 Status CLLogits1DMaxShiftExpSumKernel::validate(const ITensorInfo *input, const ITensorInfo *max, c… in validate()244 CLLogits1DMaxShiftExpSumKernel::ParallelReductionInfo CLLogits1DMaxShiftExpSumKernel::is_parallel_r… in is_parallel_reduction()251 void CLLogits1DMaxShiftExpSumKernel::run(const Window &window, cl::CommandQueue &queue) in run()
38 class CLLogits1DMaxShiftExpSumKernel; variable111 std::unique_ptr<CLLogits1DMaxShiftExpSumKernel> _max_shift_exp_sum_kernel;
45 _max_shift_exp_sum_kernel(support::cpp14::make_unique<CLLogits1DMaxShiftExpSumKernel>()), in CLSoftmaxLayerGeneric()157 …ARM_COMPUTE_RETURN_ON_ERROR(CLLogits1DMaxShiftExpSumKernel::validate(input, &tensor_info_max, &ten… in validate()
163 - @ref CLLogits1DMaxShiftExpSumKernel