Searched refs:rsum_type (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 1586 RankedTensorType rsum_type = in convertSoftmaxOp() local
1591 op->getLoc(), rsum_type, op1_exp_in.getResult(), in convertSoftmaxOp()
1594 op->getLoc(), rsum_type, op2_reducesum_op1.getResult()); in convertSoftmaxOp()
1648 RankedTensorType rsum_type = RankedTensorType::get( in convertLogSoftmaxOp() local
1652 op->getLoc(), rsum_type, op1_exp_in.getResult(), in convertLogSoftmaxOp()
1655 op->getLoc(), rsum_type, op2_reducesum_op1.getResult()); in convertLogSoftmaxOp()
|