Searched refs:num_arguments_per_3D_tensor (Results 1 – 19 of 19) sorted by relevance
161 …unsigned int idx = (1 + include_output) * num_arguments_per_3D_tensor() + 2 * num_argum… in configure()233 unsigned int idx = (1 + include_output) * num_arguments_per_3D_tensor(); in run()
86 …unsigned int idx = (1 + (_run_in_place ? 0 : 1)) * num_arguments_per_3D_tensor(); // Skip the inpu… in configure()
116 …unsigned int idx = (1 + (_run_in_place ? 0 : 1)) * num_arguments_per_3D_tensor(); // S… in configure()
121 idx = num_arguments_per_3D_tensor() * 3; in configure()
112 unsigned int idx = num_arguments_per_3D_tensor(); //Skip the tensor parameters in configure()
187 unsigned int idx = 2 * num_arguments_per_3D_tensor(); in run()
96 unsigned int idx = 2 * num_arguments_per_3D_tensor(); // Skip the src and dst parameters in configure()
130 unsigned int idx = 2 * num_arguments_per_3D_tensor(); // Skip the src and dst parameters in run_op()
143 unsigned int idx = num_arguments_per_3D_tensor() * 2; //Skip the input and output parameters in run_op()
145 unsigned int idx1 = num_arguments_per_3D_tensor(); in run_op()
144 unsigned int idx = 2 * num_arguments_per_3D_tensor(); // Skip the src and dst parameters in run_op()
144 unsigned int idx1 = num_arguments_per_3D_tensor(); in run_op()
131 unsigned int idx = num_arguments_per_3D_tensor() + num_arguments_per_2D_tensor(); in run_op()
142 unsigned int idx1 = num_arguments_per_3D_tensor(); in run_op()
416 …unsigned int idx = num_arguments_per_3D_tensor() + (_num_groups == 1 ? num_arguments_per_2D_tensor… in run_op()
265 unsigned int idx = 2 * num_arguments_per_3D_tensor(); in run_op()
220 …unsigned int idx = (in_place ? 2 : 3) * num_arguments_per_3D_tensor(); // Skip the src and dst par… in configure()
478 unsigned int idx1 = 2 * num_arguments_per_3D_tensor(); in run_op()
300 constexpr static unsigned int num_arguments_per_3D_tensor() in num_arguments_per_3D_tensor() function