OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:sum_probs
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
softmax_op_gpu.cu.cc
37
__global__ void GenerateNormalizedProb(const T* logits, const T*
sum_probs
,
49
logits[tid] - ldg(max_logits + row) - log(ldg(
sum_probs
+ row));
52
exp(logits[tid] - ldg(max_logits + row)) / ldg(
sum_probs
+ row);
107
Tensor
sum_probs
;
variable
113
softmax_out->shape(), &
sum_probs
));
134
context, const_cast<T*>(
sum_probs
.flat<T>().data()), input_itr, rows,
139
reinterpret_cast<const T*>(
sum_probs
.flat<T>().data()),
/external/opencv/ml/src/
mlem.cpp
783
CvMat*
sum_probs
= 0;
local
852
CV_CALL(
sum_probs
= cvCreateMat( 1, nclusters, CV_64FC1 ));
853
sp_data =
sum_probs
->data.db;
952
cvZero(
sum_probs
);
[
all
...]
Completed in 76 milliseconds