HomeSort by relevance Sort by last modified time
    Searched defs:batch_indices (Results 1 - 3 of 3) 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...]

Completed in 75 milliseconds