Lines Matching +full:max +full:- +full:parallel

2  * Copyright (c) 2017-2021 Arm Limited.
4  * SPDX-License-Identifier: MIT
39 /** Interface for max, shifting, exponentiating and summing the logits */
42     /**< Grid size (obtained through auto-tuning) */
44     /**< Vector size in the serial case (obtained through auto-tuning) */
46 …/**< Vector size in the parallel case (obtained through auto-tuning, enables the best memory acces…
50     /** Info for whether a parallel reduction will be run and the vector size of the execution. */
59      * @param[in,out] max             Max values tensor. Data types supported: same as @p src
64 …st CLCompileContext &compile_context, const ITensorInfo &src, ITensorInfo &max, ITensorInfo &dst, …
71 …static Status validate(const ITensorInfo &src, const ITensorInfo &max, const ITensorInfo &dst, con…
72     /** Checks if the given size is eligible for parallel reduction
75 …* @note  Parallel reduction is launched for width >= (_grid_size * _serial_vector_size) and vector…
79 …* @return A two-element tuple where the first element is a boolean specifying if a parallel reduct…
99 …* @param[in]  sum             Sum tensor. Dimensions should be dim(input)-1. Data types supported:…