Searched defs:shifted_logits (Results 1 – 2 of 2) sorted by relevance
66 auto shifted_logits = (logits - logits.maximum(along_class) in Compute() local
63 auto shifted_logits = xla::Sub(logits, logits_max, batch_dims); in Compile() local106 auto shifted_logits = xla::Sub(logits, logits_max, {kBatchDim}); in CrossEntropyWithLogits() local