OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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