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

  /external/tensorflow/tensorflow/core/kernels/
gather_nd_op_gpu.cu.cc 33 const Eigen::array<int64, IXDIM> batch_indices, const int64 indices_size,
45 out_of_bounds |= !FastBoundsCheck(index_j, batch_indices[j]);
78 Eigen::array<int64, IXDIM> batch_indices; local
81 batch_indices[size_t(IXDIM - 1)] = Tparams.dimension(IXDIM - 1);
84 batch_indices[i - 1] = Tparams.dimension(i - 1);
93 batch_indices, indices_size, s_size, out_size);
ctc_loss_op.cc 113 const int64 batch_indices = g.group()[0]; variable
114 OP_REQUIRES(ctx, FastBoundsCheck(batch_indices, batch_size),
117 " but saw: ", batch_indices));
120 std::vector<int>* b_values = &labels_t[batch_indices];
batch_kernels.cc 634 auto batch_indices = local
637 sizes.push_back(batch_indices(i, 2) - batch_indices(i, 1));
638 batch_keys.push_back(batch_indices(i, 0));
    [all...]
topk_op_gpu.cu.cc 358 auto batch_indices = indices + offset;
365 batch_indices);
  /external/tensorflow/tensorflow/python/estimator/inputs/queues/
feeding_functions.py 139 batch_indices = [
142 epoch_end_indices = [i for i, x in enumerate(batch_indices) if x == epoch_end]
146 return (batch_indices, current_epoch)
151 batch_indices = batch_indices[:final_epoch_end_inclusive + 1]
153 return (batch_indices, total_epochs)
  /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/
data_feeder.py 518 batch_indices = self.indices[self.offset:end]
522 dict([(self._input_placeholder[k].name, extract(v, batch_indices))
525 extract(self._x, batch_indices)
548 assign_label(v, shape, dtype, n_classes, batch_indices)
555 assign_label(self._y, shape, dtype, n_classes, batch_indices)

Completed in 409 milliseconds