Searched refs:tfl_softmax_op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | dequantize_tfl_softmax.cc | 49 TFL::SoftmaxOp tfl_softmax_op = cast<TFL::SoftmaxOp>(op); in matchAndRewrite() local
51 tfl_softmax_op.input().getType().cast<RankedTensorType>(); in matchAndRewrite()
55 Location loc = tfl_softmax_op.getLoc(); in matchAndRewrite()
59 loc, dequantized_input_type, tfl_softmax_op.input()); in matchAndRewrite()
61 loc, dequantized_input_type, dequantized_input, tfl_softmax_op.beta()); in matchAndRewrite()
62 Type qtype = tfl_softmax_op.output().getType(); in matchAndRewrite()
63 rewriter.replaceOpWithNewOp<TFL::QuantizeOp>(tfl_softmax_op, qtype, in matchAndRewrite()
|
| D | legalize_tfl.cc | 1952 auto tfl_softmax_op = cast<TFL::SoftmaxOp>(op); in matchAndRewrite() local
1955 rewriter, op, tfl_softmax_op.getResult(), tfl_softmax_op.input(), in matchAndRewrite()
1956 tfl_softmax_op.betaAttr().getValueAsDouble()); in matchAndRewrite()
|