HomeSort by relevance Sort by last modified time
    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