Searched refs:op1_exp_in (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 1584 auto op1_exp_in = in convertSoftmaxOp() local
1591 op->getLoc(), rsum_type, op1_exp_in.getResult(), in convertSoftmaxOp()
1597 .create<tosa::MulOp>(op->getLoc(), output_type, op1_exp_in.getResult(), in convertSoftmaxOp()
1640 auto op1_exp_in = in convertLogSoftmaxOp() local
1652 op->getLoc(), rsum_type, op1_exp_in.getResult(), in convertLogSoftmaxOp()
1658 op->getLoc(), output_type, op1_exp_in.getResult(), in convertLogSoftmaxOp()
|