Searched refs:logits_value (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 108 Value result_value, Value logits_value);
113 Value logits_value);
|
| D | legalize_common.cc | 1399 Value result_value, Value logits_value) { in convertSoftmaxOp() argument
1414 logits_value.getType().dyn_cast<RankedTensorType>(); in convertSoftmaxOp()
1452 buildRescale(rewriter, op, int32_logits_type, logits_value, 1.0f, in convertSoftmaxOp()
1585 rewriter.create<tosa::ExpOp>(op->getLoc(), output_type, logits_value); in convertSoftmaxOp()
1606 Value logits_value) { in convertLogSoftmaxOp() argument
1641 rewriter.create<tosa::ExpOp>(op->getLoc(), output_type, logits_value); in convertLogSoftmaxOp()
|