OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:along_class
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
softmax_op_functor.h
55
Eigen::DSizes<int, 1>
along_class
(kClassDim);
59
Eigen::IndexList<Eigen::type2index<kClassDim> >
along_class
;
local
67
auto shifted_logits = (logits - logits.maximum(
along_class
)
77
.sum(
along_class
)
89
softmax.device(d) = (softmax * softmax.sum(
along_class
)
xent_op.h
66
Eigen::array<int, 1>
along_class
;
local
67
along_class
[0] = kClassDim;
77
Eigen::IndexList<Eigen::type2index<kClassDim> >
along_class
;
local
87
scratch.reshape(batch_only).device(d) = logits.maximum(
along_class
);
93
scratch.reshape(batch_only).device(d) = backprop.exp().sum(
along_class
);
105
.sum(
along_class
);
sparse_xent_op.h
170
Eigen::array<int, 1>
along_class
;
local
171
along_class
[0] = kClassDim;
181
Eigen::IndexList<Eigen::type2index<kClassDim> >
along_class
;
local
191
To32Bit(scratch).device(d) = To32Bit(logits).maximum(
along_class
);
199
To32Bit(scratch).device(d) = To32Bit(backprop).exp().sum(
along_class
);
209
To32Bit(backprop).generate(sparse_xent_loss_gen).sum(
along_class
);
Completed in 67 milliseconds