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

  /external/tensorflow/tensorflow/core/kernels/
gather_functor_gpu.cu.h 38 Index indices_i = 0; local
41 indices_i = i / slice_size;
42 slice_i = i - indices_i * slice_size;
48 indices_i = batch_indices_i - batch_i * indices_size;
54 Index gather_i = ldg(indices + indices_i);
scatter_functor_gpu.cu.h 80 int indices_i = i / update_block; local
82 int param_first_index = indices[indices_i];
101 int indices_i = i / update_block; local
102 int param_first_index = indices[indices_i];
gather_nd_op_gpu.cu.cc 39 const auto indices_i = indices + IXDIM * loc; local
44 const Index index_j = ldg(indices_i + j);

Completed in 120 milliseconds