Searched refs:GenerateNormalizedProb (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | softmax_op_gpu.cu.cc | 72 __global__ void GenerateNormalizedProb(const T* logits, const U* sum_probs, in GenerateNormalizedProb() function
110 __global__ void GenerateNormalizedProb<Eigen::half, float, 8>( in GenerateNormalizedProb() function
257 GenerateNormalizedProb<T, acc_type, kUnroll>, numBlocks, in Compute()
269 GenerateNormalizedProb<T, acc_type, kUnroll>, numBlocks, in Compute()
|