HomeSort by relevance Sort by last modified time
    Searched refs:params_i (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
gather_functor_gpu.cu.h 65 Index params_i = local
67 out[i] = ldg(params + params_i);
scatter_functor_gpu.cu.h 87 int params_i = param_first_index * update_block + (i % update_block); local
88 body(&params[params_i], ldg(updates + updates_i));
108 int params_i = param_first_index * update_block + (i % update_block); local
109 body(&params[params_i], update_val);

Completed in 1023 milliseconds