Searched refs:num_arguments_per_1D_tensor (Results 1 – 12 of 12) sorted by relevance
160 … = (1 + include_output) * num_arguments_per_3D_tensor() + 2 * num_arguments_per_1D_tensor(); // S… in configure()163 idx += num_arguments_per_1D_tensor(); // Skip beta parameter in configure()167 idx += num_arguments_per_1D_tensor(); // Skip gamma parameter in configure()
93 unsigned int idx = 2 * num_arguments_per_4D_tensor() + num_arguments_per_1D_tensor(); in configure()
138 idx += (_biases != nullptr) ? num_arguments_per_1D_tensor() : 0; in run()
115 …dx = num_arguments_per_3D_tensor() + num_arguments_per_2D_tensor() + num_arguments_per_1D_tensor(); in configure()
191 unsigned int idx = _has_same_rank ? 0 : num_arguments_per_1D_tensor(); in run()
252 …dx2 = 2 * num_arguments_per_4D_tensor() + ((_bias != nullptr) ? num_arguments_per_1D_tensor() : 0); in run()
534 … = 3 * num_arguments_per_3D_tensor() + ((_biases != nullptr) ? num_arguments_per_1D_tensor() : 0) … in configure()
81 unsigned int num_arguments_per_1D_tensor() const;
90 idx += (biases != nullptr) ? num_arguments_per_1D_tensor() : 0; in configure()
325 …dx = (_bias == nullptr) ? 3 * num_arguments_per_3D_tensor() : (num_arguments_per_1D_tensor() + 3 *… in configure()
137 unsigned int IGCKernel::num_arguments_per_1D_tensor() const in num_arguments_per_1D_tensor() function in IGCKernel
186 constexpr static unsigned int num_arguments_per_1D_tensor() in num_arguments_per_1D_tensor() function