Searched refs:num_arguments_per_2D_tensor (Results 1 – 10 of 10) sorted by relevance
113 idx = num_arguments_per_2D_tensor() * 3; in configure()117 idx = num_arguments_per_2D_tensor() * 3; in configure()
177 _kernel.setArg(num_arguments_per_2D_tensor() * num_tensors, local_res_size, nullptr); in run()
180 unsigned int idx = num_arguments_per_2D_tensor(); in configure()
398 … idx0 = (4 + _num_post_op_args) * num_arguments_per_2D_tensor() + (7 + _num_post_op_args); in run_op()402 … idx0 = (3 + _num_post_op_args) * num_arguments_per_2D_tensor() + (6 + _num_post_op_args); in run_op()414 …idx0 = (4 + _num_post_op_args) * num_arguments_per_2D_tensor() + 7 + (_reinterpret_input_as_3d ? 1… in run_op()418 …idx0 = (3 + _num_post_op_args) * num_arguments_per_2D_tensor() + 6 + (_reinterpret_input_as_3d ? 1… in run_op()
302 const unsigned int idx0 = 3 * num_arguments_per_2D_tensor() + 3; in run_op()310 …const unsigned int idx0 = 3 * num_arguments_per_2D_tensor() + 3 + (_reinterpret_i… in run_op()
131 unsigned int idx = num_arguments_per_3D_tensor() + num_arguments_per_2D_tensor(); in run_op()
485 const unsigned int idx0 = 3 * num_arguments_per_2D_tensor() + 3; in run_op()493 …const unsigned int idx0 = 3 * num_arguments_per_2D_tensor() + 3 + (_reinterpret_i… in run_op()
274 const unsigned int idx0 = 3 * num_arguments_per_2D_tensor() + 4; in run_op()
416 … int idx = num_arguments_per_3D_tensor() + (_num_groups == 1 ? num_arguments_per_2D_tensor() : num… in run_op()
292 constexpr static unsigned int num_arguments_per_2D_tensor() in num_arguments_per_2D_tensor() function