OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
full:batch_only
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
xent_op.h
68
Eigen::array<int, 1>
batch_only
;
local
69
batch_only
[0] = batch_size;
80
Eigen::IndexList<int>
batch_only
;
local
81
batch_only
.set(0, batch_size);
87
scratch.reshape(
batch_only
).device(d) = logits.maximum(along_class);
93
scratch.reshape(
batch_only
).device(d) = backprop.exp().sum(along_class);
sparse_xent_op.h
172
Eigen::array<int, 1>
batch_only
;
local
173
batch_only
[0] = batch_size;
184
Eigen::IndexList<int>
batch_only
;
local
185
batch_only
.set(0, batch_size);
Completed in 135 milliseconds