Searched refs:shifted_logits (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | softmax_op_functor.h | 66 auto shifted_logits = (logits - logits.maximum(along_class) in Compute() local
73 softmax.device(d) = shifted_logits; in Compute()
86 softmax.device(d) = shifted_logits.exp(); in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 88 auto shifted_logits = xla::Sub(logits, logits_max, batch_dims); in Compile() local
89 auto exp_shifted = xla::Exp(shifted_logits); in Compile()
103 ? xla::Sub(shifted_logits, xla::Log(sum), batch_dims) in Compile()
131 auto shifted_logits = xla::Sub(logits, logits_max, {kBatchDim}); in CrossEntropyWithLogits() local
134 auto exp_shifted_logits = xla::Exp(shifted_logits); in CrossEntropyWithLogits()
152 auto sub = xla::Sub(shifted_logits, log_sum_exp, {kBatchDim}); in CrossEntropyWithLogits()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | ctc_loss_op_test.py | 506 shifted_logits = array_ops.concat([
517 logits=shifted_logits,
566 shifted_logits = array_ops.concat([
576 logits=shifted_logits,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2729 auto shifted_logits = in matchAndRewrite() local
2733 Value exp = rewriter.create<ExpOp>(loc, type, shifted_logits); in matchAndRewrite()
2744 rewriter.replaceOpWithNewOp<mhlo::SubOp>(op, shifted_logits, in matchAndRewrite()
|