Searched refs:max_logits (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | softmax_op_gpu.cu.cc | 73 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()
|