Home
last modified time | relevance | path

Searched refs:max_logits (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsoftmax_op_gpu.cu.cc73 const T* max_logits, T* output, in GenerateNormalizedProb() argument
83 U max_val = strict_cast<U>(ldg(max_logits + row)); in GenerateNormalizedProb()
99 const T* max_logits, in SubtractAndExpFunctor()
101 : logits_(logits), max_logits_(max_logits), num_cols_(num_cols) {} in SubtractAndExpFunctor()
149 Tensor max_logits; in Compute() local
153 softmax_out->shape(), &max_logits)); in Compute()
161 context, const_cast<T*>(max_logits.flat<T>().data()), in Compute()
177 reinterpret_cast<const T*>(max_logits.flat<T>().data()), cols)); in Compute()
187 reinterpret_cast<const T*>(max_logits.flat<T>().data()), in Compute()