HomeSort by relevance Sort by last modified time
    Searched refs:slice_i (Results 1 - 4 of 4) 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/_impl/keras/utils/
training_utils.py 182 slice_i = Lambda(
190 inputs.append(slice_i)
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
gather_op.cc 138 auto slice_i = bodyb->Reshape(
149 auto updated_output = bodyb->DynamicUpdateSlice(output, slice_i, out_index);
  /external/tensorflow/tensorflow/compiler/xla/
array.h 425 int64 slice_i = 0;
431 result.values_[slice_i++] = values_[i];

Completed in 345 milliseconds