Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsoftmax_op_gpu.cu.cc73 const T* max_logits, T* output, in GenerateNormalizedProb()
112 const Eigen::half* max_logits, Eigen::half* output, const int num_rows, in GenerateNormalizedProb()
215 Tensor max_logits; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlower_tf.cc1667 auto max_logits = in matchAndRewrite() local