Home
last modified time | relevance | path

Searched refs:num_arguments_per_1D_tensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLBatchNormalizationLayerKernel.cpp161 … = (1 + include_output) * num_arguments_per_3D_tensor() + 2 * num_arguments_per_1D_tensor(); // S… in configure()
164 idx += num_arguments_per_1D_tensor(); // Skip beta parameter in configure()
168 idx += num_arguments_per_1D_tensor(); // Skip gamma parameter in configure()
H A DCLReverseKernel.cpp95 unsigned int idx = 2 * num_arguments_per_4D_tensor() + num_arguments_per_1D_tensor(); in configure()
H A DCLSelectKernel.cpp154 unsigned int idx = _has_same_rank ? 0 : num_arguments_per_1D_tensor(); in run()
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClWeightsReshapeKernel.cpp132 idx += (biases != nullptr) ? num_arguments_per_1D_tensor() : 0; in run_op()
H A DClWinogradOutputTransformKernel.cpp295 …idx2 = 2 * num_arguments_per_4D_tensor() + ((bias != nullptr) ? num_arguments_per_1D_tensor() : 0); in run_op()
/aosp_15_r20/external/ComputeLibrary/src/core/CL/
H A DICLKernel.h284 constexpr static unsigned int num_arguments_per_1D_tensor() in num_arguments_per_1D_tensor() function