Home
last modified time | relevance | path

Searched refs:ClLogits1DMaxShiftExpSumKernel (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClSoftmaxKernel.cpp151 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::ClLogits1DMaxShiftExpSumKernel
162 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()
H A DClSoftmaxKernel.h40 class ClLogits1DMaxShiftExpSumKernel : public IClKernel
53 ClLogits1DMaxShiftExpSumKernel();
54 ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(ClLogits1DMaxShiftExpSumKernel);
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClSoftmax.h40 class ClLogits1DMaxShiftExpSumKernel; variable
80 std::unique_ptr<kernels::ClLogits1DMaxShiftExpSumKernel> _max_shift_exp_sum_kernel;
H A DClSoftmax.cpp44 _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()