Searched refs:ClLogits1DMaxShiftExpSumKernel (Results 1 – 4 of 4) sorted by relevance
151 const unsigned int ClLogits1DMaxShiftExpSumKernel::_grid_size = 64;153 const unsigned int ClLogits1DMaxShiftExpSumKernel::_serial_vector_size = 8;155 const unsigned int ClLogits1DMaxShiftExpSumKernel::_parallel_vector_size = 4;157 ClLogits1DMaxShiftExpSumKernel::ClLogits1DMaxShiftExpSumKernel() in ClLogits1DMaxShiftExpSumKernel() function in arm_compute::opencl::kernels::ClLogits1DMaxShiftExpSumKernel162 void ClLogits1DMaxShiftExpSumKernel::configure(const CLCompileContext &compile_context, const ITens… in configure()212 Status ClLogits1DMaxShiftExpSumKernel::validate(const ITensorInfo &src, const ITensorInfo &max, con… in validate()218 ClLogits1DMaxShiftExpSumKernel::ParallelReductionInfo ClLogits1DMaxShiftExpSumKernel::is_parallel_r… in is_parallel_reduction()225 void ClLogits1DMaxShiftExpSumKernel::run_op(ITensorPack &tensors, const Window &window, ::cl::Comma… in run_op()
40 class ClLogits1DMaxShiftExpSumKernel : public IClKernel53 ClLogits1DMaxShiftExpSumKernel();54 ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(ClLogits1DMaxShiftExpSumKernel);
40 class ClLogits1DMaxShiftExpSumKernel; variable80 std::unique_ptr<kernels::ClLogits1DMaxShiftExpSumKernel> _max_shift_exp_sum_kernel;
44 _max_shift_exp_sum_kernel(std::make_unique<kernels::ClLogits1DMaxShiftExpSumKernel>()), in ClSoftmax()127 …ARM_COMPUTE_RETURN_ON_ERROR(kernels::ClLogits1DMaxShiftExpSumKernel::validate(src, tensor_info_max… in validate()