Searched refs:params_i (Results 1 – 2 of 2) sorted by relevance
87 int params_i = param_first_index * update_block + (i % update_block);88 body(¶ms[params_i], ldg(updates + updates_i));108 int params_i = param_first_index * update_block + (i % update_block);109 body(¶ms[params_i], update_val);
65 Index params_i = in GatherOpKernel() local67 out[i] = ldg(params + params_i); in GatherOpKernel()