Searched refs:num_arguments_per_1D_tensor (Results 1 – 6 of 6) sorted by relevance
161 … = (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()
95 unsigned int idx = 2 * num_arguments_per_4D_tensor() + num_arguments_per_1D_tensor(); in configure()
154 unsigned int idx = _has_same_rank ? 0 : num_arguments_per_1D_tensor(); in run()
132 idx += (biases != nullptr) ? num_arguments_per_1D_tensor() : 0; in run_op()
295 …idx2 = 2 * num_arguments_per_4D_tensor() + ((bias != nullptr) ? num_arguments_per_1D_tensor() : 0); in run_op()
284 constexpr static unsigned int num_arguments_per_1D_tensor() in num_arguments_per_1D_tensor() function