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

  /external/tensorflow/tensorflow/core/kernels/
gather_functor_gpu.cu.h 39 Index slice_i = 0; local
42 slice_i = i - indices_i * slice_size;
50 slice_i = i - batch_indices_i * slice_size;
63 // params[batch_i, gather_i, slice_i] and write it to the i'th position in
66 (batch_i * gather_dim_size + gather_i) * slice_size + slice_i;
  /external/tensorflow/tensorflow/python/keras/utils/
multi_gpu_utils.py 227 slice_i = Lambda(
235 inputs.append(slice_i)
  /external/tensorflow/tensorflow/compiler/xla/
array.h 458 int64 slice_i = 0;
464 result.values_[slice_i++] = values_[i];

Completed in 108 milliseconds